SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[run_slow] idefics2 (#32840)

Browse Source
This commit is contained in:
Andrés Marafioti
2024-08-22 18:08:03 +02:00
committed by GitHub
parent 975b988bfe
commit 18199b34e5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/idefics2/test_modeling_idefics2.py
View File

@@ -34,6 +34,7 @@ from transformers.testing_utils import (
require_flash_attn, require_flash_attn,
require_torch, require_torch,
require_torch_gpu, require_torch_gpu,
require_torch_multi_gpu,
slow, slow,
torch_device, torch_device,
) )
@@ -498,7 +499,7 @@ class Idefics2ForConditionalGenerationIntegrationTest(unittest.TestCase):
torch.cuda.empty_cache() torch.cuda.empty_cache()
@slow @slow
@unittest.skip("Test hits OOM on CI - https://github.com/huggingface/transformers/issues/32288") @require_torch_multi_gpu
def test_integration_test(self): def test_integration_test(self):
model = Idefics2ForConditionalGeneration.from_pretrained( model = Idefics2ForConditionalGeneration.from_pretrained(
"HuggingFaceM4/idefics2-8b-base", "HuggingFaceM4/idefics2-8b-base",