From 4ecd19a92c538171d170f28f6ee38c458f2d9800 Mon Sep 17 00:00:00 2001 From: Alex-Brooks Date: Mon, 28 Oct 2024 19:46:28 -0400 Subject: [PATCH] Remove large GPU marker from llava one vision tests Signed-off-by: Alex-Brooks --- tests/models/decoder_only/vision_language/test_models.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/models/decoder_only/vision_language/test_models.py b/tests/models/decoder_only/vision_language/test_models.py index 8a79cfea528e8..f2d39b6577a7a 100644 --- a/tests/models/decoder_only/vision_language/test_models.py +++ b/tests/models/decoder_only/vision_language/test_models.py @@ -207,7 +207,6 @@ ), limit_mm_per_prompt={"video": 4}, )], - marks=[large_gpu_mark(min_gb=48)], ), "llava_next_video": VLMTestInfo( models=["llava-hf/LLaVA-NeXT-Video-7B-hf"], @@ -364,7 +363,6 @@ ), limit_mm_per_prompt={"image": 4}, )], - marks=[large_gpu_mark(min_gb=48)], ), } # yapf: enable