Add core_model mark for Qwen2-VL embedding-as-input tests

Signed-off-by: imkero <[email protected]>
vllm-project · Nov 13, 2024 · ee2344b · ee2344b
1 parent 6d1a8e5
commit ee2344b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/models/decoder_only/vision_language/test_qwen2_vl.py b/tests/models/decoder_only/vision_language/test_qwen2_vl.py
@@ -297,6 +297,7 @@ def run_test(
         )
 
 
+@pytest.mark.core_model
 @pytest.mark.parametrize("model", models)
 @pytest.mark.parametrize(
     "size_factors",
@@ -337,6 +338,7 @@ def test_qwen2_vl_image_embeddings_input(vllm_runner, image_assets, model,
     )
 
 
+@pytest.mark.core_model
 @pytest.mark.parametrize("model", models)
 @pytest.mark.parametrize(
     "size_factors",
@@ -381,6 +383,7 @@ def test_qwen2_vl_multiple_image_embeddings_input(vllm_runner, image_assets,
     )
 
 
+@pytest.mark.core_model
 @pytest.mark.parametrize("model", models)
 @pytest.mark.parametrize(
     "size_factors",