mirror of https://github.com/coqui-ai/TTS.git
49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
#!/usr/bin/env python3`
|
|
import glob
|
|
import os
|
|
import shutil
|
|
|
|
from tests import get_tests_output_path, run_cli
|
|
from TTS.tts.utils.speakers import SpeakerManager
|
|
from TTS.utils.generic_utils import get_user_data_dir
|
|
from TTS.utils.manage import ModelManager
|
|
|
|
|
|
def test_run_all_models():
|
|
"""Check if all the models are downloadable and tts models run correctly."""
|
|
print(" > Run synthesizer with all the models.")
|
|
download_dir = get_user_data_dir("tts")
|
|
output_path = os.path.join(get_tests_output_path(), "output.wav")
|
|
manager = ModelManager(output_prefix=get_tests_output_path())
|
|
model_names = manager.list_models()
|
|
for model_name in model_names:
|
|
model_path, _, _ = manager.download_model(model_name)
|
|
if "tts_models" in model_name:
|
|
local_download_dir = os.path.dirname(model_path)
|
|
# download and run the model
|
|
speaker_files = glob.glob(local_download_dir + "/speaker*")
|
|
if len(speaker_files) > 0:
|
|
# multi-speaker model
|
|
if "speaker_ids" in speaker_files[0]:
|
|
speaker_manager = SpeakerManager(speaker_id_file_path=speaker_files[0])
|
|
elif "speakers" in speaker_files[0]:
|
|
speaker_manager = SpeakerManager(d_vectors_file_path=speaker_files[0])
|
|
speaker_id = list(speaker_manager.speaker_ids.keys())[0]
|
|
run_cli(
|
|
f"tts --model_name {model_name} "
|
|
f'--text "This is an example." --out_path "{output_path}" --speaker_idx "{speaker_id}"'
|
|
)
|
|
else:
|
|
# single-speaker model
|
|
run_cli(f"tts --model_name {model_name} " f'--text "This is an example." --out_path "{output_path}"')
|
|
# remove downloaded models
|
|
shutil.rmtree(download_dir)
|
|
else:
|
|
# only download the model
|
|
manager.download_model(model_name)
|
|
print(f" | > OK: {model_name}")
|
|
|
|
folders = glob.glob(os.path.join(manager.output_prefix, "*"))
|
|
assert len(folders) == len(model_names)
|
|
shutil.rmtree(manager.output_prefix)
|