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

Merge pull request from coqui-ai/merge_2651 

Resolve conflicts
This commit is contained in:
Eren Gölge 2023-07-06 09:53:48 +02:00 committed by GitHub
parent e42a72eb79 53938e2d32
commit 08bc758cad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions
TTS/utils
manage.py

19
TTS/utils/manage.py
View File

@ -317,14 +317,19 @@ class ModelManager(object):
else: else:
os.makedirs(output_path, exist_ok=True) os.makedirs(output_path, exist_ok=True)
print(f" > Downloading model to {output_path}") print(f" > Downloading model to {output_path}")
if "fairseq" in model_name: try:
self.download_fairseq_model(model_name, output_path) if "fairseq" in model_name:
elif "github_rls_url" in model_item: self.download_fairseq_model(model_name, output_path)
self._download_github_model(model_item, output_path) elif "github_rls_url" in model_item:
elif "hf_url" in model_item: self._download_github_model(model_item, output_path)
self._download_hf_model(model_item, output_path) elif "hf_url" in model_item:
self._download_hf_model(model_item, output_path)
self.print_model_license(model_item=model_item) except requests.Exception.RequestException as e:
print(f" > Failed to download the model file to {output_path}")
rmtree(output_path)
raise e
self.print_model_license(model_item=model_item)
# find downloaded files # find downloaded files
output_model_path = output_path output_model_path = output_path
output_config_path = None output_config_path = None