Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add xtts to zoo test exception

This commit is contained in:
WeberJulian 2023-10-20 06:15:22 -03:00
parent d07b37b19c
commit 5e7cba22bd
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/zoo_tests/test_models.py
View File

@ -15,6 +15,7 @@ MODELS_WITH_SEP_TESTS = [
"tts_models/multilingual/multi-dataset/bark", "tts_models/multilingual/multi-dataset/bark",
"tts_models/en/multi-dataset/tortoise-v2", "tts_models/en/multi-dataset/tortoise-v2",
"tts_models/multilingual/multi-dataset/xtts_v1", "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"' f'--speaker_wav "{speaker_wav}" --language_idx "en"'
) )
def test_xtts_streaming(): def test_xtts_streaming():
"""Testing the new inference_stream method""" """Testing the new inference_stream method"""
from TTS.tts.configs.xtts_config import XttsConfig from TTS.tts.configs.xtts_config import XttsConfig
@ -122,6 +124,7 @@ def test_xtts_streaming():
wav_chuncks.append(chunk) wav_chuncks.append(chunk)
assert len(wav_chuncks) > 1 assert len(wav_chuncks) > 1
def test_tortoise(): def test_tortoise():
output_path = os.path.join(get_tests_output_path(), "output.wav") output_path = os.path.join(get_tests_output_path(), "output.wav")
use_gpu = torch.cuda.is_available() use_gpu = torch.cuda.is_available()