Skip to content

Commit

Permalink
Remove large GPU marker from llava one vision tests
Browse files Browse the repository at this point in the history
Signed-off-by: Alex-Brooks <[email protected]>
  • Loading branch information
alex-jw-brooks committed Oct 28, 2024
1 parent 7f1c5dd commit 4ecd19a
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tests/models/decoder_only/vision_language/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"],
Expand Down Expand Up @@ -364,7 +363,6 @@
),
limit_mm_per_prompt={"image": 4},
)],
marks=[large_gpu_mark(min_gb=48)],
),
}
# yapf: enable
Expand Down

0 comments on commit 4ecd19a

Please sign in to comment.