Remove large gpu test decorator

vllm-project · Oct 28, 2024 · b28ebc8 · b28ebc8
1 parent f831f9e
commit b28ebc8
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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])