diff --git a/tests/zoo_tests/test_models.py b/tests/zoo_tests/test_models.py index dc16d793..db144f1c 100644 --- a/tests/zoo_tests/test_models.py +++ b/tests/zoo_tests/test_models.py @@ -15,6 +15,7 @@ MODELS_WITH_SEP_TESTS = [ "tts_models/multilingual/multi-dataset/bark", "tts_models/en/multi-dataset/tortoise-v2", "tts_models/multilingual/multi-dataset/xtts_v1", + "tts_models/multilingual/multi-dataset/xtts_v1.1", ] @@ -93,6 +94,7 @@ def test_xtts(): f'--speaker_wav "{speaker_wav}" --language_idx "en"' ) + def test_xtts_streaming(): """Testing the new inference_stream method""" from TTS.tts.configs.xtts_config import XttsConfig @@ -122,6 +124,7 @@ def test_xtts_streaming(): wav_chuncks.append(chunk) assert len(wav_chuncks) > 1 + def test_tortoise(): output_path = os.path.join(get_tests_output_path(), "output.wav") use_gpu = torch.cuda.is_available()