Fix fairseq (#11)

* fix fairseq mode

* Added line to fix fairseq model issue and made code cleaner.

---------

Co-authored-by: akgupta1337 <akgupta1337@gmail.com>
This commit is contained in:
Enno Hermann 2024-03-09 16:43:42 +01:00 committed by GitHub
parent 2e8f47a33d
commit d80f7f4eba
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions

View File

@ -260,8 +260,7 @@ 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]
model_type, lang, dataset, model = model_name.split("/")
model_item = {
"model_type": "tts_models",
"license": "CC BY-NC 4.0",