Add neon models (#2140)

* Add neon ljspeech vits model

* Add neon german model

* Update .models.json

* Add neon spanish model

* Add french model

* Add Dutch model

* Add Hungarian model

* Add Greek model

* Remove uneeded description

* Update .models.json

* Update .models.json

* Handling neon models

* Add all neon models

* Update .models.json

* Split zoo_tests

* Update test names

* Update model testing

Co-authored-by: Eren Gölge <erogol@hotmail.com>
This commit is contained in:
logan hart
2022-11-16 16:12:39 +01:00
committed by GitHub
co-authored by Eren Gölge
parent a0f31df481
commit ff9b63d02a
6 changed files with 385 additions and 14 deletions
+18 -7
View File
@@ -10,14 +10,13 @@ from TTS.utils.generic_utils import get_user_data_dir
from TTS.utils.manage import ModelManager
def test_run_all_models():
def run_models(offset=0, step=1):
"""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(), progress_bar=False)
model_names = manager.list_models()
for model_name in model_names:
for model_name in model_names[offset::step]:
print(f"\n > Run - {model_name}")
model_path, _, _ = manager.download_model(model_name)
if "tts_models" in model_name:
@@ -50,15 +49,27 @@ def test_run_all_models():
f'--text "This is an example." --out_path "{output_path}" --progress_bar False'
)
# remove downloaded models
shutil.rmtree(download_dir)
shutil.rmtree(local_download_dir)
shutil.rmtree(get_user_data_dir("tts"))
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)
# folders = glob.glob(os.path.join(manager.output_prefix, "*"))
# assert len(folders) == len(model_names) // step
def test_models_offset_0_step_3():
run_models(offset=0, step=3)
def test_models_offset_1_step_3():
run_models(offset=1, step=3)
def test_models_offset_2_step_3():
run_models(offset=2, step=3)
def test_voice_conversion():