From 87e405dff22270ada33735f36a3acda8073741e5 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Thu, 28 Sep 2023 10:33:08 -0300 Subject: [PATCH] Add Tortoise inference unit test --- tests/zoo_tests/test_models.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) 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")