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

Update manage.py 

fix: fairseq model
This commit is contained in:
David Martin Rius 2024-03-05 18:31:08 +01:00 committed by GitHub
parent ea3ae40888
commit 64fdd0ed8b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/utils/manage.py
View File

@ -260,10 +260,13 @@ class ModelManager(object):
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
if "fairseq" in model_name: if "fairseq" in model_name:
model_type = "tts_models" split = model_name.split("/")
lang = model_name.split("/")[1] model_type = split[0]
lang = split[1]
dataset = split[2]
model = split[3]
model_item = { model_item = {
"model_type": "tts_models", "model_type": model_type,
"license": "CC BY-NC 4.0", "license": "CC BY-NC 4.0",
"default_vocoder": None, "default_vocoder": None,
"author": "fairseq", "author": "fairseq",