From ee2344b9ce300f962498f61a5f8c73b921e941c2 Mon Sep 17 00:00:00 2001 From: imkero Date: Wed, 13 Nov 2024 12:30:42 +0800 Subject: [PATCH] Add core_model mark for Qwen2-VL embedding-as-input tests Signed-off-by: imkero --- tests/models/decoder_only/vision_language/test_qwen2_vl.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/models/decoder_only/vision_language/test_qwen2_vl.py b/tests/models/decoder_only/vision_language/test_qwen2_vl.py index 5354b854aacaa..718c675b86fb4 100644 --- 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",