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

Fix XTTS tests

This commit is contained in:
WeberJulian 2023-11-14 14:26:54 +01:00
parent b3b22a04ca
commit f20fa19efa
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/zoo_tests/test_models.py
View File

@ -111,7 +111,7 @@ def test_xtts_streaming():
model.to(torch.device("cuda" if torch.cuda.is_available() else "cpu")) model.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
print("Computing speaker latents...") print("Computing speaker latents...")
gpt_cond_latent, _, speaker_embedding = model.get_conditioning_latents(audio_path=speaker_wav) gpt_cond_latent, speaker_embedding = model.get_conditioning_latents(audio_path=speaker_wav)
print("Inference...") print("Inference...")
chunks = model.inference_stream( chunks = model.inference_stream(
@ -139,7 +139,7 @@ def test_xtts_v2():
"yes | " "yes | "
f"tts --model_name tts_models/multilingual/multi-dataset/xtts_v2 " f"tts --model_name tts_models/multilingual/multi-dataset/xtts_v2 "
f'--text "This is an example." --out_path "{output_path}" --progress_bar False --use_cuda True ' f'--text "This is an example." --out_path "{output_path}" --progress_bar False --use_cuda True '
f'--speaker_wav "{speaker_wav}" "{speaker_wav_2}" "--language_idx "en"' f'--speaker_wav "{speaker_wav}" "{speaker_wav_2}" --language_idx "en"'
) )
else: else:
run_cli( run_cli(
@ -164,7 +164,7 @@ def test_xtts_v2_streaming():
model.to(torch.device("cuda" if torch.cuda.is_available() else "cpu")) model.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
print("Computing speaker latents...") print("Computing speaker latents...")
gpt_cond_latent, _, speaker_embedding = model.get_conditioning_latents(audio_path=speaker_wav) gpt_cond_latent, speaker_embedding = model.get_conditioning_latents(audio_path=speaker_wav)
print("Inference...") print("Inference...")
chunks = model.inference_stream( chunks = model.inference_stream(