pull/2945/head
Eren G??lge 2023-09-13 18:21:44 +02:00
parent 5dcf9ae311
commit ee7cee0e35
2 changed files with 8 additions and 0 deletions

View File

@ -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)

View File

@ -48,3 +48,5 @@ einops
transformers
#deps for bark
encodec
# deps for XTTS
unidecode