[run_slow] idefics2 (huggingface#32840)

zucchini-nlp · Aug 30, 2024 · c344102 · c344102
1 parent 8c0f4dc
commit c344102
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/models/idefics2/test_modeling_idefics2.py b/tests/models/idefics2/test_modeling_idefics2.py
@@ -34,6 +34,7 @@
     require_flash_attn,
     require_torch,
     require_torch_gpu,
+    require_torch_multi_gpu,
     slow,
     torch_device,
 )
@@ -498,7 +499,7 @@ def tearDown(self):
         torch.cuda.empty_cache()
 
     @slow
-    @unittest.skip("Test hits OOM on CI - https://github.com/huggingface/transformers/issues/32288")
+    @require_torch_multi_gpu
     def test_integration_test(self):
         model = Idefics2ForConditionalGeneration.from_pretrained(
             "HuggingFaceM4/idefics2-8b-base",