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

Add more listing options to ModelManager

This commit is contained in:
Eren Gölge 2021-12-20 11:53:44 +00:00
parent a25269d897
commit f769595112
1 changed files with 51 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
TTS/utils/manage.py
View File

@ -46,36 +46,66 @@ class ModelManager(object):
with open(file_path, "r", encoding="utf-8") as json_file: with open(file_path, "r", encoding="utf-8") as json_file:
self.models_dict = json.load(json_file) self.models_dict = json.load(json_file)
def list_langs(self): def _list_models(self, model_type, model_count=0):
print(" Name format: type/language") model_list = []
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]:
print(f" >: {model_type}/{lang} ") for model in self.models_dict[model_type][lang][dataset]:
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_count}: {model_type}/{lang}/{dataset}/{model} [already downloaded]")
else:
print(f" {model_count}: {model_type}/{lang}/{dataset}/{model}")
model_list.append(f"{model_type}/{lang}/{dataset}/{model}")
model_count += 1
return model_list
def list_datasets(self): def _list_for_model_type(self, model_type):
print(" Name format: type/language/dataset") print(" Name format: language/dataset/model")
for model_type in self.models_dict: models_name_list = []
for lang in self.models_dict[model_type]: model_count = 1
for dataset in self.models_dict[model_type][lang]: model_type = "tts_models"
print(f" >: {model_type}/{lang}/{dataset}") models_name_list.extend(self._list_models(model_type, model_count))
return [name.replace(model_type + "/", "") for name in models_name_list]
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 model_count = 1
for model_type in self.models_dict:
model_list = self._list_models(model_type, model_count)
models_name_list.extend(model_list)
return models_name_list
def list_tts_models(self):
"""Print all `TTS` models and return a list of model names
Format is `language/dataset/model`
"""
return self._list_for_model_type("tts_models")
def list_vocoder_models(self):
"""Print all the `vocoder` models and return a list of model names
Format is `language/dataset/model`
"""
return self._list_for_model_type("vocoder_models")
def list_langs(self):
"""Print all the available languages"""
print(" Name format: type/language")
for model_type in self.models_dict:
for lang in self.models_dict[model_type]:
print(f" >: {model_type}/{lang} ")
def list_datasets(self):
"""Print all the datasets"""
print(" Name format: type/language/dataset")
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]:
for model in self.models_dict[model_type][lang][dataset]: print(f" >: {model_type}/{lang}/{dataset}")
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_count}: {model_type}/{lang}/{dataset}/{model} [already downloaded]")
else:
print(f" {model_count}: {model_type}/{lang}/{dataset}/{model}")
models_name_list.append(f"{model_type}/{lang}/{dataset}/{model}")
model_count += 1
return models_name_list
def download_model(self, model_name): def download_model(self, model_name):
"""Download model files given the full model name. """Download model files given the full model name.