From d0ab51ce1c068686f7459dae0a48cccf26e89402 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9s=20Marafioti?= <andimarafioti@gmail.com>
Date: Thu, 22 Aug 2024 18:08:03 +0200
Subject: [PATCH] [run_slow] idefics2 (#32840)

---
 tests/models/idefics2/test_modeling_idefics2.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/models/idefics2/test_modeling_idefics2.py b/tests/models/idefics2/test_modeling_idefics2.py
index d1a982dc595cb1..8d48acb9500d2f 100644
--- 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",