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

View File

@ -260,10 +260,13 @@ class ModelManager(object):
def _set_model_item(self, model_name):
# fetch model info from the dict
if "fairseq" in model_name:
model_type = "tts_models"
lang = model_name.split("/")[1]
split = model_name.split("/")
model_type = split[0]
lang = split[1]
dataset = split[2]
model = split[3]
model_item = {
"model_type": "tts_models",
"model_type": model_type,
"license": "CC BY-NC 4.0",
"default_vocoder": None,
"author": "fairseq",