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 manage

This commit is contained in:
Eren G??lge 2023-06-22 10:13:20 +02:00
parent fff8b762bc
commit e888e8a56d
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
TTS/utils/manage.py
View File

@ -264,19 +264,21 @@ class ModelManager(object):
model_download_uri = os.path.join(URI_PREFIX, f"{lang}.tar.gz") model_download_uri = os.path.join(URI_PREFIX, f"{lang}.tar.gz")
self._download_tar_file(model_download_uri, output_path, self.progress_bar) self._download_tar_file(model_download_uri, output_path, self.progress_bar)
def set_model_url(self, model_item: Dict): @staticmethod
def set_model_url(model_item: Dict):
model_item["model_url"] = None model_item["model_url"] = None
if "github_rls_url" in model_item: if "github_rls_url" in model_item:
model_item["model_url"] = model_item["github_rls_url"] model_item["model_url"] = model_item["github_rls_url"]
elif "hf_url" in model_item: elif "hf_url" in model_item:
model_item["model_url"] = model_item["hf_url"] model_item["model_url"] = model_item["hf_url"]
elif "fairseq" in model_item["model_name"]:
model_item["model_url"] = "https://coqui.gateway.scarf.sh/fairseq/"
return model_item return model_item
def _set_model_item(self, model_name): def _set_model_item(self, model_name):
# fetch model info from the dict # fetch model info from the dict
model_type, lang, dataset, model = model_name.split("/") model_type, lang, dataset, model = model_name.split("/")
model_full_name = f"{model_type}--{lang}--{dataset}--{model}" model_full_name = f"{model_type}--{lang}--{dataset}--{model}"
model_item = self.set_model_url(model_item)
if "fairseq" in model_name: if "fairseq" in model_name:
model_item = { model_item = {
"model_type": "tts_models", "model_type": "tts_models",
@ -289,6 +291,7 @@ class ModelManager(object):
# get model from models.json # get model from models.json
model_item = self.models_dict[model_type][lang][dataset][model] model_item = self.models_dict[model_type][lang][dataset][model]
model_item["model_type"] = model_type model_item["model_type"] = model_type
model_item = self.set_model_url(model_item)
return model_item, model_full_name, model return model_item, model_full_name, model
def download_model(self, model_name): def download_model(self, model_name):