diff --git a/tests/models/embedding/language/test_cls_models.py b/tests/models/embedding/language/test_cls_models.py index 6321503e7b248..f39d6aef00d90 100644 --- a/tests/models/embedding/language/test_cls_models.py +++ b/tests/models/embedding/language/test_cls_models.py @@ -1,7 +1,4 @@ -"""Compare the outputs of HF and vLLM when using greedy sampling. - -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. +"""Compare the classification outputs of HF and vLLM models. Run `pytest tests/models/test_cls_models.py`. """ @@ -13,6 +10,8 @@ @pytest.mark.parametrize( "model", [ + pytest.param("ai21labs/Jamba-tiny-reward-dev", + marks=[pytest.mark.core_model]), pytest.param("jason9693/Qwen2.5-1.5B-apeach", marks=[pytest.mark.core_model, pytest.mark.cpu_model]), ], diff --git a/tests/models/embedding/language/test_scoring.py b/tests/models/embedding/language/test_scoring.py index af31e1a635f65..be6e3842821e2 100644 --- a/tests/models/embedding/language/test_scoring.py +++ b/tests/models/embedding/language/test_scoring.py @@ -1,6 +1,6 @@ -"""Compare the embedding outputs of HF and vLLM models. +"""Compare the scoring outputs of HF and vLLM models. -Run `pytest tests/models/embedding/language/test_embedding.py`. +Run `pytest tests/models/embedding/language/test_scoring.py`. """ import math