Bug fix on XTTS v2.0 Trainer

pull/3149/head
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

View File

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

View File

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