gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-29 14:15:18 -08:00
Code Issues Packages Projects Releases Wiki Activity

change work dir to F5TTS

Browse Source
This commit is contained in:
ZhikangNiu
2024-11-28 00:33:30 +08:00
parent 65a649c683
commit f18c715056
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/f5_tts/train/train.py
View File

@@ -8,8 +8,10 @@ from f5_tts.model import CFM, DiT, Trainer, UNetT
from f5_tts.model.dataset import load_dataset
from f5_tts.model.utils import get_tokenizer
os.chdir(str(files("f5_tts").joinpath("../..")))
@hydra.main(config_path=os.path.join("..", "configs"), config_name=None)
@hydra.main(version_base="1.3", config_path=str(files("f5_tts").joinpath("configs")), config_name=None)
def main(cfg):
tokenizer = cfg.model.tokenizer
mel_spec_type = cfg.model.mel_spec.mel_spec_type