[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_phi3v.py
+++ b/tests/models/embedding/vision_language/test_phi3v.py
@@ -74,6 +74,7 @@ def _run_test(
    )


+@pytest.mark.core_model
@pytest.mark.parametrize("model", MODELS)
@pytest.mark.parametrize("dtype", ["half"])
 def test_models_text(
@@ -98,6 +99,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(