Skip to content

Commit 18199b3

Browse files
[run_slow] idefics2 (#32840)
1 parent 975b988 commit 18199b3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/models/idefics2/test_modeling_idefics2.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
require_flash_attn,
3535
require_torch,
3636
require_torch_gpu,
37+
require_torch_multi_gpu,
3738
slow,
3839
torch_device,
3940
)
@@ -498,7 +499,7 @@ def tearDown(self):
498499
torch.cuda.empty_cache()
499500

500501
@slow
501-
@unittest.skip("Test hits OOM on CI - https://github.com/huggingface/transformers/issues/32288")
502+
@require_torch_multi_gpu
502503
def test_integration_test(self):
503504
model = Idefics2ForConditionalGeneration.from_pretrained(
504505
"HuggingFaceM4/idefics2-8b-base",

0 commit comments

Comments
 (0)