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

Fix seamless TTS generate (#34968)

Browse Source 
* fix seamless tts generate

* apply same fix for v2

* [run-slow] seamless_m4t, seamless_m4t_v2

* remove TODO

* [run-slow] seamless_m4t, seamless_m4t_v2

* [run-slow] seamless_m4t, seamless_m4t_v2

* ignore failing test on multigpus

* [run-slow] seamless_m4t, seamless_m4t_v2

* [run-slow] seamless_m4t, seamless_m4t_v2
This commit is contained in:
Yoach Lacombe
2024-12-11 15:38:42 +01:00
committed by GitHub
parent 33c12e4d80
commit 6181c6b095
4 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/seamless_m4t_v2/test_modeling_seamless_m4t_v2.py
View File

@@ -589,6 +589,11 @@ class SeamlessM4Tv2ModelWithSpeechInputTest(ModelTesterMixin, unittest.TestCase)
[self.model_tester.num_attention_heads, encoder_seq_length, encoder_key_length],
)
# TODO: @ydshieh: refer to #34968
@unittest.skip(reason="Failing on multi-gpu runner")
def test_retain_grad_hidden_states_attentions(self):
pass
@require_torch
class SeamlessM4Tv2ModelWithTextInputTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):