From fee6ffe6ec66cecdfdbb13f17a49d87ddd1e9f6e Mon Sep 17 00:00:00 2001 From: Antony Date: Sat, 6 Jan 2024 23:16:18 -0300 Subject: [PATCH] fix: fairseq model --- TTS/utils/manage.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py index 3a527f4609..5284005388 100644 --- 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",