diff --git a/tests/models/decoder_only/vision_language/test_llava_onevision.py b/tests/models/decoder_only/vision_language/test_llava_onevision.py index 4e07f41555623..1616fd299b9aa 100644 --- a/tests/models/decoder_only/vision_language/test_llava_onevision.py +++ b/tests/models/decoder_only/vision_language/test_llava_onevision.py @@ -11,7 +11,6 @@ from ....conftest import (VIDEO_ASSETS, HfRunner, PromptImageInput, PromptVideoInput, VllmRunner) -from ....utils import large_gpu_test from ...utils import check_logprobs_close # Video test @@ -113,7 +112,6 @@ def process(hf_inputs: BatchEncoding): ) -@large_gpu_test(min_gb=48) @pytest.mark.parametrize("model", models) @pytest.mark.parametrize("dtype", ["half"]) @pytest.mark.parametrize("max_tokens", [128]) @@ -229,7 +227,6 @@ def process(hf_inputs: BatchEncoding): ) -@large_gpu_test(min_gb=48) @pytest.mark.parametrize("model", models) @pytest.mark.parametrize("dtype", ["half"]) @pytest.mark.parametrize("max_tokens", [128])