Skip to content

Commit

Permalink
Add Tortoise inference unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
Edresson committed Sep 28, 2023
1 parent 01c04f7 commit 87e405d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion tests/zoo_tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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")
Expand Down

0 comments on commit 87e405d

Please sign in to comment.