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

Print model count when listing modelsk

This commit is contained in:
Eren Gölge 2021-08-10 16:25:11 +00:00
parent 09ed8426e8
commit 537bc8487a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/utils/manage.py
View File

@ -64,6 +64,7 @@ class ModelManager(object):
def list_models(self): def list_models(self):
print(" Name format: type/language/dataset/model") print(" Name format: type/language/dataset/model")
models_name_list = [] models_name_list = []
model_count = 1
for model_type in self.models_dict: for model_type in self.models_dict:
for lang in self.models_dict[model_type]: for lang in self.models_dict[model_type]:
for dataset in self.models_dict[model_type][lang]: for dataset in self.models_dict[model_type][lang]:
@ -71,10 +72,11 @@ class ModelManager(object):
model_full_name = 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) output_path = os.path.join(self.output_prefix, model_full_name)
if os.path.exists(output_path): if os.path.exists(output_path):
print(f" >: {model_type}/{lang}/{dataset}/{model} [already downloaded]") print(f" {model_count}: {model_type}/{lang}/{dataset}/{model} [already downloaded]")
else: else:
print(f" >: {model_type}/{lang}/{dataset}/{model}") print(f" {model_count}: {model_type}/{lang}/{dataset}/{model}")
models_name_list.append(f"{model_type}/{lang}/{dataset}/{model}") models_name_list.append(f"{model_type}/{lang}/{dataset}/{model}")
model_count += 1
return models_name_list return models_name_list
def download_model(self, model_name): def download_model(self, model_name):