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

Bug fix on XTTS v2.0 Trainer

This commit is contained in:
Edresson Casanova 2023-11-06 15:22:18 -03:00 committed by Eren G??lge
parent f444f296f2
commit b146de4ce8
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/layers/xtts/trainer/gpt_trainer.py
View File

@ -237,8 +237,7 @@ class GPTTrainer(BaseTTS):
self.config, self.config,
s_info["speaker_wav"], s_info["speaker_wav"],
s_info["language"], s_info["language"],
gpt_cond_len=3, gpt_cond_len=3
decoder="ne_hifigan",
)["wav"] )["wav"]
test_audios["{}-audio".format(idx)] = wav test_audios["{}-audio".format(idx)] = wav

2
tests/xtts_tests/test_xtts_v2-0_gpt_train.py
View File

@ -87,7 +87,9 @@ model_args = GPTArgs(
gpt_use_masking_gt_prompt_approach=True, gpt_use_masking_gt_prompt_approach=True,
gpt_use_perceiver_resampler=True, gpt_use_perceiver_resampler=True,
) )
audio_config = XttsAudioConfig(sample_rate=22050, dvae_sample_rate=22050, output_sample_rate=24000) audio_config = XttsAudioConfig(sample_rate=22050, dvae_sample_rate=22050, output_sample_rate=24000)
config = GPTTrainerConfig( config = GPTTrainerConfig(
epochs=1, epochs=1,
output_path=OUT_PATH, output_path=OUT_PATH,