Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixup

This commit is contained in:
Eren G??lge 2023-09-13 18:21:44 +02:00
parent 5dcf9ae311
commit ee7cee0e35
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/config/__init__.py
View File

@ -37,6 +37,11 @@ def register_config(model_name: str) -> Coqpit:
""" """
config_class = None config_class = None
config_name = model_name + "_config" 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"] paths = ["TTS.tts.configs", "TTS.vocoder.configs", "TTS.encoder.configs", "TTS.vc.configs"]
for path in paths: for path in paths:
try: try:
@ -91,6 +96,7 @@ def load_config(config_path: str) -> Coqpit:
raise TypeError(f" [!] Unknown config file type {ext}") raise TypeError(f" [!] Unknown config file type {ext}")
config_dict.update(data) config_dict.update(data)
model_name = _process_model_name(config_dict) model_name = _process_model_name(config_dict)
breakpoint
config_class = register_config(model_name.lower()) config_class = register_config(model_name.lower())
config = config_class() config = config_class()
config.from_dict(config_dict) config.from_dict(config_dict)

2
requirements.txt
View File

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