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

Flag SpeechT5 flaky test (#37587)

Browse Source 
flag flaky test
This commit is contained in:
Pablo Montalvo
2025-04-18 11:35:46 +02:00
committed by GitHub
parent aa17cfb4d5
commit 049b75ea72
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/speecht5/test_modeling_speecht5.py
View File

@@ -20,6 +20,7 @@ import unittest
from transformers import SpeechT5Config, SpeechT5HifiGanConfig from transformers import SpeechT5Config, SpeechT5HifiGanConfig
from transformers.testing_utils import ( from transformers.testing_utils import (
is_flaky,
is_torch_available, is_torch_available,
require_deterministic_for_xpu, require_deterministic_for_xpu,
require_sentencepiece, require_sentencepiece,
@@ -722,6 +723,10 @@ class SpeechT5ForSpeechToTextTest(ModelTesterMixin, unittest.TestCase, Generatio
def test_training_gradient_checkpointing_use_reentrant_false(self): def test_training_gradient_checkpointing_use_reentrant_false(self):
pass pass
@is_flaky(max_attempts=5, description="Flaky for some input configurations.")
def test_past_key_values_format(self):
super().test_past_key_values_format()
# overwrite from test_modeling_common # overwrite from test_modeling_common
def _mock_init_weights(self, module): def _mock_init_weights(self, module):
if hasattr(module, "weight") and module.weight is not None: if hasattr(module, "weight") and module.weight is not None: