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

fix(utils.manage): remove bare except, improve messages

This commit is contained in:
Enno Hermann 2024-03-31 13:21:01 +02:00
parent 7dc5d1eb3d
commit e689fd1d4a
1 changed files with 36 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
TTS/utils/manage.py
View File

@ -97,21 +97,36 @@ class ModelManager(object):
models_name_list.extend(model_list) models_name_list.extend(model_list)
return models_name_list return models_name_list
def log_model_details(self, model_type, lang, dataset, model):
logger.info("Model type: %s", model_type)
logger.info("Language supported: %s", lang)
logger.info("Dataset used: %s", dataset)
logger.info("Model name: %s", model)
if "description" in self.models_dict[model_type][lang][dataset][model]:
logger.info("Description: %s", self.models_dict[model_type][lang][dataset][model]["description"])
else:
logger.info("Description: coming soon")
if "default_vocoder" in self.models_dict[model_type][lang][dataset][model]:
logger.info(
"Default vocoder: %s",
self.models_dict[model_type][lang][dataset][model]["default_vocoder"],
)
def model_info_by_idx(self, model_query): def model_info_by_idx(self, model_query):
"""Print the description of the model from .models.json file using model_idx """Print the description of the model from .models.json file using model_query_idx
Args: Args:
model_query (str): <model_tye>/<model_idx> model_query (str): <model_tye>/<model_query_idx>
""" """
model_name_list = [] model_name_list = []
model_type, model_query_idx = model_query.split("/") model_type, model_query_idx = model_query.split("/")
try: try:
model_query_idx = int(model_query_idx) model_query_idx = int(model_query_idx)
if model_query_idx <= 0: if model_query_idx <= 0:
print("> model_query_idx should be a positive integer!") logger.error("model_query_idx [%d] should be a positive integer!", model_query_idx)
return return
except: except (TypeError, ValueError):
print("> model_query_idx should be an integer!") logger.error("model_query_idx [%s] should be an integer!", model_query_idx)
return return
model_count = 0 model_count = 0
if model_type in self.models_dict: if model_type in self.models_dict:
@ -121,22 +136,13 @@ class ModelManager(object):
model_name_list.append(f"{model_type}/{lang}/{dataset}/{model}") model_name_list.append(f"{model_type}/{lang}/{dataset}/{model}")
model_count += 1 model_count += 1
else: else:
print(f"> model_type {model_type} does not exist in the list.") logger.error("Model type %s does not exist in the list.", model_type)
return return
if model_query_idx > model_count: if model_query_idx > model_count:
print(f"model query idx exceeds the number of available models [{model_count}] ") logger.error("model_query_idx exceeds the number of available models [%d]", model_count)
else: else:
model_type, lang, dataset, model = model_name_list[model_query_idx - 1].split("/") model_type, lang, dataset, model = model_name_list[model_query_idx - 1].split("/")
print(f"> model type : {model_type}") self.log_model_details(model_type, lang, dataset, model)
print(f"> language supported : {lang}")
print(f"> dataset used : {dataset}")
print(f"> model name : {model}")
if "description" in self.models_dict[model_type][lang][dataset][model]:
print(f"> description : {self.models_dict[model_type][lang][dataset][model]['description']}")
else:
print("> description : coming soon")
if "default_vocoder" in self.models_dict[model_type][lang][dataset][model]:
print(f"> default_vocoder : {self.models_dict[model_type][lang][dataset][model]['default_vocoder']}")
def model_info_by_full_name(self, model_query_name): def model_info_by_full_name(self, model_query_name):
"""Print the description of the model from .models.json file using model_full_name """Print the description of the model from .models.json file using model_full_name
@ -145,32 +151,19 @@ class ModelManager(object):
model_query_name (str): Format is <model_type>/<language>/<dataset>/<model_name> model_query_name (str): Format is <model_type>/<language>/<dataset>/<model_name>
""" """
model_type, lang, dataset, model = model_query_name.split("/") model_type, lang, dataset, model = model_query_name.split("/")
if model_type in self.models_dict: if model_type not in self.models_dict:
if lang in self.models_dict[model_type]: logger.error("Model type %s does not exist in the list.", model_type)
if dataset in self.models_dict[model_type][lang]: return
if model in self.models_dict[model_type][lang][dataset]: if lang not in self.models_dict[model_type]:
print(f"> model type : {model_type}") logger.error("Language %s does not exist for %s.", lang, model_type)
print(f"> language supported : {lang}") return
print(f"> dataset used : {dataset}") if dataset not in self.models_dict[model_type][lang]:
print(f"> model name : {model}") logger.error("Dataset %s does not exist for %s/%s.", dataset, model_type, lang)
if "description" in self.models_dict[model_type][lang][dataset][model]: return
print( if model not in self.models_dict[model_type][lang][dataset]:
f"> description : {self.models_dict[model_type][lang][dataset][model]['description']}" logger.error("Model %s does not exist for %s/%s/%s.", model, model_type, lang, dataset)
) return
else: self.log_model_details(model_type, lang, dataset, model)
print("> description : coming soon")
if "default_vocoder" in self.models_dict[model_type][lang][dataset][model]:
print(
f"> default_vocoder : {self.models_dict[model_type][lang][dataset][model]['default_vocoder']}"
)
else:
print(f"> model {model} does not exist for {model_type}/{lang}/{dataset}.")
else:
print(f"> dataset {dataset} does not exist for {model_type}/{lang}.")
else:
print(f"> lang {lang} does not exist for {model_type}.")
else:
print(f"> model_type {model_type} does not exist in the list.")
def list_tts_models(self): def list_tts_models(self):
"""Print all `TTS` models and return a list of model names """Print all `TTS` models and return a list of model names