We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 975b988 commit 18199b3Copy full SHA for 18199b3
tests/models/idefics2/test_modeling_idefics2.py
@@ -34,6 +34,7 @@
34
require_flash_attn,
35
require_torch,
36
require_torch_gpu,
37
+ require_torch_multi_gpu,
38
slow,
39
torch_device,
40
)
@@ -498,7 +499,7 @@ def tearDown(self):
498
499
torch.cuda.empty_cache()
500
501
@slow
- @unittest.skip("Test hits OOM on CI - https://github.com/huggingface/transformers/issues/32288")
502
+ @require_torch_multi_gpu
503
def test_integration_test(self):
504
model = Idefics2ForConditionalGeneration.from_pretrained(
505
"HuggingFaceM4/idefics2-8b-base",
0 commit comments