fix: fairseq model

coqui-ai · Jan 7, 2024 · fee6ffe · fee6ffe
1 parent 5dcc16d
commit fee6ffe
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py
@@ -260,10 +260,13 @@ 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]
+            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",