Bug fix on XTTS v2.0 Trainer

coqui-ai · Nov 6, 2023 · b146de4 · b146de4
1 parent f444f29
commit b146de4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/TTS/tts/layers/xtts/trainer/gpt_trainer.py b/TTS/tts/layers/xtts/trainer/gpt_trainer.py
@@ -237,8 +237,7 @@ def test_run(self, assets) -> Tuple[Dict, Dict]: # pylint: disable=W0613
  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
 

diff --git a/tests/xtts_tests/test_xtts_v2-0_gpt_train.py b/tests/xtts_tests/test_xtts_v2-0_gpt_train.py
@@ -87,7 +87,9 @@
  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,