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

SeamlessM4T: test_retain_grad_hidden_states_attentions is flaky (#28035)

Browse Source
This commit is contained in:
Joao Gante
2023-12-14 13:56:03 +00:00
committed by GitHub
parent 9e5c28c573
commit e2b16485f3
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/seamless_m4t/test_modeling_seamless_m4t.py
View File

@@ -20,7 +20,7 @@ import tempfile
import unittest import unittest
from transformers import SeamlessM4TConfig, is_speech_available, is_torch_available from transformers import SeamlessM4TConfig, is_speech_available, is_torch_available
from transformers.testing_utils import require_torch, slow, torch_device from transformers.testing_utils import is_flaky, require_torch, slow, torch_device
from transformers.trainer_utils import set_seed from transformers.trainer_utils import set_seed
from transformers.utils import cached_property from transformers.utils import cached_property
@@ -610,6 +610,10 @@ class SeamlessM4TModelWithSpeechInputTest(ModelTesterMixin, unittest.TestCase):
[self.model_tester.num_attention_heads, encoder_seq_length, encoder_key_length], [self.model_tester.num_attention_heads, encoder_seq_length, encoder_key_length],
) )
@is_flaky()
def test_retain_grad_hidden_states_attentions(self):
super().test_retain_grad_hidden_states_attentions()
@require_torch @require_torch
class SeamlessM4TModelWithTextInputTest( class SeamlessM4TModelWithTextInputTest(