fix missing dataset, model var initialization

2023-12-27 01:16:05 -05:00 · 2023-12-27 01:16:05 -05:00 · 169146184e
parent 55c7063724
commit 169146184e
1 changed files with 2 additions and 0 deletions
--- a/TTS/utils/manage.py
+++ b/TTS/utils/manage.py
@ -262,6 +262,8 @@ class ModelManager(object):
        if "fairseq" in model_name:
            model_type = "tts_models"
            lang = model_name.split("/")[1]
+            dataset = "fairseq"
+            model = "vits"
            model_item = {
                "model_type": "tts_models",
                "license": "CC BY-NC 4.0",