From ccbd542eb07a8349eeeecc975bca33258252be9e Mon Sep 17 00:00:00 2001 From: Thorsten Mueller Date: Wed, 27 Jan 2021 16:19:02 +0100 Subject: [PATCH] Added info if model already downloaded in --list_models --- TTS/utils/manage.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py index 25b3d797..af741156 100644 --- a/TTS/utils/manage.py +++ b/TTS/utils/manage.py @@ -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.