[Model] Support Qwen2 embeddings and use tags to select model tests (#10184)

2024-11-15 12:23:09 +08:00
parent 2885ba0e24
commit b40cf6402e
19 changed files with 252 additions and 178 deletions
--- a/tests/models/embedding/vision_language/test_llava_next.py
+++ b/tests/models/embedding/vision_language/test_llava_next.py
@@ -88,6 +88,7 @@ def _run_test(

@pytest.mark.skipif(transformers.__version__.startswith("4.46"),
                    reason="Model broken with changes in transformers 4.46")
+@pytest.mark.core_model
@pytest.mark.parametrize("model", MODELS)
@pytest.mark.parametrize("dtype", ["half"])
 def test_models_text(
@@ -112,6 +113,7 @@ def test_models_text(


@large_gpu_test(min_gb=48)
+@pytest.mark.core_model
@pytest.mark.parametrize("model", MODELS)
@pytest.mark.parametrize("dtype", ["half"])
 def test_models_image(