diff --git a/Makefile b/Makefile index 6b9fcf00ec..61e59b660d 100644 --- a/Makefile +++ b/Makefile @@ -146,7 +146,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 637fa9fa2d..38140576bc 100644 --- a/tests/test_image_to_text_example.py +++ b/tests/test_image_to_text_example.py @@ -32,8 +32,8 @@ "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-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), ], }