From d7de111b76a9f141f8f29df1ef401ff8b0588703 Mon Sep 17 00:00:00 2001 From: WeberJulian Date: Wed, 1 Jun 2022 10:31:26 +0200 Subject: [PATCH] Change default speakers file extension --- TTS/tts/models/base_tts.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/TTS/tts/models/base_tts.py b/TTS/tts/models/base_tts.py index c71872d3..c86bd391 100644 --- a/TTS/tts/models/base_tts.py +++ b/TTS/tts/models/base_tts.py @@ -407,16 +407,16 @@ class BaseTTS(BaseTrainerModel): return test_figures, test_audios def on_init_start(self, trainer): - """Save the speaker.json and language_ids.json at the beginning of the training. Also update both paths.""" + """Save the speaker.pth and language_ids.json at the beginning of the training. Also update both paths.""" if self.speaker_manager is not None: - output_path = os.path.join(trainer.output_path, "speakers.json") + output_path = os.path.join(trainer.output_path, "speakers.pth") self.speaker_manager.save_ids_to_file(output_path) trainer.config.speakers_file = output_path # some models don't have `model_args` set if hasattr(trainer.config, "model_args"): trainer.config.model_args.speakers_file = output_path trainer.config.save_json(os.path.join(trainer.output_path, "config.json")) - print(f" > `speakers.json` is saved to {output_path}.") + print(f" > `speakers.pth` is saved to {output_path}.") print(" > `speakers_file` is updated in the config.json.") if hasattr(self, "language_manager") and self.language_manager is not None: