riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats 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.testing_utils import (
is_flaky,
is_torch_available,
require_deterministic_for_xpu,
require_sentencepiece,
@ -722,6 +723,10 @@ class SpeechT5ForSpeechToTextTest(ModelTesterMixin, unittest.TestCase, Generatio
def test_training_gradient_checkpointing_use_reentrant_false(self):
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
def _mock_init_weights(self, module):
if hasattr(module, "weight") and module.weight is not None: