From 64fdd0ed8b792af8bbd21b95024c8f31da314ac4 Mon Sep 17 00:00:00 2001 From: David Martin Rius <0991592@gmail.com> Date: Tue, 5 Mar 2024 18:31:08 +0100 Subject: [PATCH] Update manage.py 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 3a527f46..52840053 100644 --- a/TTS/utils/manage.py +++ b/TTS/utils/manage.py @@ -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",