Fix fairseq (#11)

* fix fairseq mode * Added line to fix fairseq model issue and made code cleaner. --------- Co-authored-by: akgupta1337 <[email protected]>
eginhard · Mar 9, 2024 · d80f7f4 · d80f7f4
1 parent 2e8f47a
commit d80f7f4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py
@@ -260,8 +260,7 @@ def set_model_url(model_item: Dict):
     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",