fix: fairseq model

This commit is contained in:
Antony 2024-01-06 23:16:18 -03:00 committed by GitHub
parent 5dcc16d193
commit fee6ffe6ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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",