diff --git a/tests/models/decoder_only/language/test_cls_models.py b/tests/models/decoder_only/language/test_cls_models.py index daf6461b38dcb..d8ca6d361f0e3 100644 --- a/tests/models/decoder_only/language/test_cls_models.py +++ b/tests/models/decoder_only/language/test_cls_models.py @@ -3,7 +3,7 @@ This test only tests small models. Big models such as 7B should be tested from test_big_models.py because it could use a larger instance to run tests. -Run `pytest tests/models/test_models.py`. +Run `pytest tests/models/test_cls_models.py`. """ import pytest import torch