We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f1ec48 commit 1efc64aCopy full SHA for 1efc64a
1 file changed
tests/test_huggingface_vlm.py
@@ -20,9 +20,10 @@
20
def gemma3():
21
return HuggingFaceVLM(
22
model_id=HuggingFaceVLMs.gemma_3_27b_it,
23
- use_hosted_model=True,
+ use_hosted_model=False,
24
# 10 frames gets OOM at A100 (80GB) VRAM.
25
- max_n_frames_per_video=5,
+ # 4 frames gets OOM at L4 (24GB) VRAM.
26
+ max_n_frames_per_video=3,
27
)
28
29
@@ -59,7 +60,7 @@ def test_huggingface_vlm_multi_turn_with_images(gemma3):
59
60
assert_model_supports_multiturn_with_multiple_imgs(gemma3)
61
62
-@pytest.mark.skipif(condition=is_ci(), reason="Files are not available on CI")
63
+@pytest.mark.skip(reason="We currently run OOM on serverless and hosted.")
64
def test_huggingface_vlm_multi_turn_with_6min_video(gemma3):
65
assert_model_supports_multiturn_with_6min_video(gemma3)
66
0 commit comments