mirror of https://github.com/coqui-ai/TTS.git
Fixup
parent
5dcf9ae311
commit
ee7cee0e35
|
@ -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)
|
||||
|
|
|
@ -48,3 +48,5 @@ einops
|
|||
transformers
|
||||
#deps for bark
|
||||
encodec
|
||||
# deps for XTTS
|
||||
unidecode
|
||||
|
|
Loading…
Reference in New Issue