From 537bc8487a4eb3d260864ac9c990dfb6d8b2414d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 10 Aug 2021 16:25:11 +0000 Subject: [PATCH] Print model count when listing modelsk --- TTS/utils/manage.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py index d5e8d410..1d61d392 100644 --- a/TTS/utils/manage.py +++ b/TTS/utils/manage.py @@ -64,6 +64,7 @@ class ModelManager(object): def list_models(self): print(" Name format: type/language/dataset/model") models_name_list = [] + model_count = 1 for model_type in self.models_dict: for lang in self.models_dict[model_type]: 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}" 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]") + print(f" {model_count}: {model_type}/{lang}/{dataset}/{model} [already downloaded]") 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}") + model_count += 1 return models_name_list def download_model(self, model_name):