From 05ebe8b9b0011734f1294dc583ac41410783b48c Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 19 Nov 2024 17:32:10 +0100 Subject: [PATCH] Run `test_medium_seamless_m4t_pt` in `subprocess` to avoid many failures (#34812) * fix * fix * fix --------- Co-authored-by: ydshieh --- tests/pipelines/test_pipelines_text_to_audio.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/pipelines/test_pipelines_text_to_audio.py b/tests/pipelines/test_pipelines_text_to_audio.py index e07e2ad392a3e6..dac2ce6b30ec22 100644 --- a/tests/pipelines/test_pipelines_text_to_audio.py +++ b/tests/pipelines/test_pipelines_text_to_audio.py @@ -27,6 +27,7 @@ require_torch, require_torch_accelerator, require_torch_or_tf, + run_test_using_subprocess, slow, torch_device, ) @@ -66,8 +67,10 @@ def test_small_musicgen_pt(self): audio = [output["audio"] for output in outputs] self.assertEqual([ANY(np.ndarray), ANY(np.ndarray)], audio) + # TODO: @ylacombe: `SeamlessM4TForTextToSpeech.generate` has issue with `generation_config`. See issue #34811 @slow @require_torch + @run_test_using_subprocess def test_medium_seamless_m4t_pt(self): speech_generator = pipeline(task="text-to-audio", model="facebook/hf-seamless-m4t-medium", framework="pt")