remove model from 4bit test

huggingface · Aug 15, 2024 · 0f7a8e6 · 0f7a8e6
1 parent e4cbe65
commit 0f7a8e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/models/idefics3/test_modeling_idefics3.py b/tests/models/idefics3/test_modeling_idefics3.py
@@ -486,6 +486,7 @@ def tearDown(self):
         torch.cuda.empty_cache()
 
     @slow
+    @unittest.skip("Test hits OOM on CI - Same as idefics2 - https://github.com/huggingface/transformers/issues/32288")
     def test_integration_test(self):
         model = Idefics3ForConditionalGeneration.from_pretrained(
             "HuggingFaceM4/Idefics3-8B-Llama3",
@@ -510,7 +511,7 @@ def test_integration_test(self):
     def test_integration_test_4bit(self):
         # Let' s make sure we test the preprocessing to replace what is used
         model = Idefics3ForConditionalGeneration.from_pretrained(
-            "HuggingFaceM4/Idefics3-8B-Llama3", load_in_4bit=True, device_map="auto"
+            "HuggingFaceM4/Idefics3-8B-Llama3", load_in_4bit=True
         )
 
         # Create pixel inputs