Fixup

coqui-ai · Sep 13, 2023 · ee7cee0 · ee7cee0
1 parent 5dcf9ae
commit ee7cee0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/TTS/config/__init__.py b/TTS/config/__init__.py
@@ -37,6 +37,11 @@ def register_config(model_name: str) -> Coqpit:
     """
     config_class = None
     config_name = model_name + "_config"
+
+    # TODO: fix this
+    if model_name == "xtts":
+        from TTS.tts.configs.xtts_config import XttsConfig
+        config_class = XttsConfig
     paths = ["TTS.tts.configs", "TTS.vocoder.configs", "TTS.encoder.configs", "TTS.vc.configs"]
     for path in paths:
         try:
@@ -91,6 +96,7 @@ def load_config(config_path: str) -> Coqpit:
         raise TypeError(f" [!] Unknown config file type {ext}")
     config_dict.update(data)
     model_name = _process_model_name(config_dict)
+    breakpoint
     config_class = register_config(model_name.lower())
     config = config_class()
     config.from_dict(config_dict)

diff --git a/requirements.txt b/requirements.txt
@@ -48,3 +48,5 @@ einops
 transformers
 #deps for bark
 encodec
+# deps for XTTS
+unidecode