diff --git a/TTS/config/__init__.py b/TTS/config/__init__.py index 49ac8f4a..b9200cd0 100644 --- 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 index 5712c2c8..ae22b333 100644 --- a/requirements.txt +++ b/requirements.txt @@ -48,3 +48,5 @@ einops transformers #deps for bark encodec +# deps for XTTS +unidecode