wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added info if model already downloaded in --list_models

Browse Source
This commit is contained in:
Thorsten Mueller
2021-02-02 19:29:16 +01:00
parent df5899daf4
commit c75ea74914
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/utils/manage.py
+6 -1
View File
@@ -52,7 +52,12 @@ class ModelManager(object):
for lang in self.models_dict[model_type]:
for dataset in self.models_dict[model_type][lang]:
for model in self.models_dict[model_type][lang][dataset]:
print(f" >: {model_type}/{lang}/{dataset}/{model} ")
model_full_name = f"{model_type}--{lang}--{dataset}--{model}"
output_path = os.path.join(self.output_prefix, model_full_name)
if os.path.exists(output_path):
print(f" >: {model_type}/{lang}/{dataset}/{model} [already downloaded]")
else:
print(f" >: {model_type}/{lang}/{dataset}/{model}")
def download_model(self, model_name):
"""Download model files given the full model name.