From 8916a7187ae6e65cc0f2e99bedfee992c708c59c Mon Sep 17 00:00:00 2001 From: gplutop7 Date: Mon, 12 May 2025 16:59:21 +0300 Subject: [PATCH 1/2] add groups 1x, 2x and 8x to slow_tests_image_to_text_example --- Makefile | 5 ++++- tests/test_image_to_text_example.py | 6 +++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index 406afbdc51..f276677f18 100644 --- a/Makefile +++ b/Makefile @@ -145,7 +145,10 @@ slow_tests_image_to_text_example: test_installs python -m pytest tests/test_image_to_text_example.py -v -s --token $(TOKEN) slow_tests_image_to_text_example_1x: test_installs - python -m pytest tests/test_image_to_text_example.py -m "not x8" -v -s --token $(TOKEN) + python -m pytest tests/test_image_to_text_example.py -m "(not x2) and (not x8)" -v -s --token $(TOKEN) + +slow_tests_image_to_text_example_2x: test_installs + python -m pytest tests/test_image_to_text_example.py -m x2 -v -s --token $(TOKEN) slow_tests_image_to_text_example_8x: test_installs python -m pytest tests/test_image_to_text_example.py -m x8 -v -s --token $(TOKEN) diff --git a/tests/test_image_to_text_example.py b/tests/test_image_to_text_example.py index de805dffd2..c62a1ddd07 100644 --- a/tests/test_image_to_text_example.py +++ b/tests/test_image_to_text_example.py @@ -32,9 +32,9 @@ "fp8": [ # ("llava-hf/llava-1.5-7b-hf", 1), # ("llava-hf/llava-1.5-13b-hf", 1), - ("llava-hf/llava-v1.6-mistral-7b-hf", 1), - ("llava-hf/llava-v1.6-vicuna-7b-hf", 1), - pytest.param("llava-hf/llava-v1.6-vicuna-13b-hf", 1, marks=pytest.mark.x8), + ("llava-hf/llava-v1.6-mistral-7b-hf", 1, marks=pytest.mark.x2), + ("llava-hf/llava-v1.6-vicuna-7b-hf", 1, marks=pytest.mark.x2), + ("llava-hf/llava-v1.6-vicuna-13b-hf", 1, marks=pytest.mark.x8), ], } else: From 1622733a84324fb2db0b64a28d3b7f361c95cdea Mon Sep 17 00:00:00 2001 From: gplutop7 Date: Mon, 12 May 2025 17:06:43 +0300 Subject: [PATCH 2/2] add groups 1x, 2x and 8x to slow_tests_image_to_text_example - part.2 --- tests/test_image_to_text_example.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_image_to_text_example.py b/tests/test_image_to_text_example.py index c62a1ddd07..f58e097d33 100644 --- a/tests/test_image_to_text_example.py +++ b/tests/test_image_to_text_example.py @@ -32,9 +32,9 @@ "fp8": [ # ("llava-hf/llava-1.5-7b-hf", 1), # ("llava-hf/llava-1.5-13b-hf", 1), - ("llava-hf/llava-v1.6-mistral-7b-hf", 1, marks=pytest.mark.x2), - ("llava-hf/llava-v1.6-vicuna-7b-hf", 1, marks=pytest.mark.x2), - ("llava-hf/llava-v1.6-vicuna-13b-hf", 1, marks=pytest.mark.x8), + pytest.param("llava-hf/llava-v1.6-mistral-7b-hf", 1, marks=pytest.mark.x2), + pytest.param("llava-hf/llava-v1.6-vicuna-7b-hf", 1, marks=pytest.mark.x2), + pytest.param("llava-hf/llava-v1.6-vicuna-13b-hf", 1, marks=pytest.mark.x8), ], } else: