Skip to content

Commit

Permalink
Run test_medium_seamless_m4t_pt in subprocess to avoid many failu…
Browse files Browse the repository at this point in the history
…res (#34812)

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <[email protected]>
  • Loading branch information
ydshieh and ydshieh authored Nov 19, 2024
1 parent eedc113 commit 05ebe8b
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/pipelines/test_pipelines_text_to_audio.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
require_torch,
require_torch_accelerator,
require_torch_or_tf,
run_test_using_subprocess,
slow,
torch_device,
)
Expand Down Expand Up @@ -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")

Expand Down

0 comments on commit 05ebe8b

Please sign in to comment.