diff --git a/tests/zoo_tests/test_models.py b/tests/zoo_tests/test_models.py index b7945d6e82..932c6ab5cf 100644 --- a/tests/zoo_tests/test_models.py +++ b/tests/zoo_tests/test_models.py @@ -9,7 +9,7 @@ from TTS.utils.generic_utils import get_user_data_dir from TTS.utils.manage import ModelManager -MODELS_WITH_SEP_TESTS = ["bark", "xtts"] +MODELS_WITH_SEP_TESTS = ["bark", "tortoise", "xtts"] def run_models(offset=0, step=1): @@ -77,6 +77,14 @@ def test_xtts(): ) +def test_tortoise(): + output_path = os.path.join(get_tests_output_path(), "output.wav") + run_cli( + f" tts --model_name tts_models/en/multi-dataset/tortoise-v2 " + f'--text "This is an example." --out_path "{output_path}" --progress_bar False --use_cuda True' + ) + + def test_bark(): """Bark is too big to run on github actions. We need to test it locally""" output_path = os.path.join(get_tests_output_path(), "output.wav")