From 8c32a6998a7a6f2ad47f54ca1dc1ece29ae235d8 Mon Sep 17 00:00:00 2001 From: Eren G??lge Date: Mon, 26 Dec 2022 14:29:25 +0100 Subject: [PATCH] Add pth files to manager --- TTS/utils/manage.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py index 33011921..44348b29 100644 --- a/TTS/utils/manage.py +++ b/TTS/utils/manage.py @@ -298,7 +298,9 @@ class ModelManager(object): """ output_stats_path = os.path.join(output_path, "scale_stats.npy") output_d_vector_file_path = os.path.join(output_path, "speakers.json") + output_d_vector_file_pth_path = os.path.join(output_path, "speakers.pth") output_speaker_ids_file_path = os.path.join(output_path, "speaker_ids.json") + output_speaker_ids_file_pth_path = os.path.join(output_path, "speaker_ids.pth") speaker_encoder_config_path = os.path.join(output_path, "config_se.json") speaker_encoder_model_path = self._find_speaker_encoder(output_path) @@ -307,11 +309,15 @@ class ModelManager(object): # update the speakers.json file path in the model config.json to the current path self._update_path("d_vector_file", output_d_vector_file_path, config_path) + self._update_path("d_vector_file", output_d_vector_file_pth_path, config_path) self._update_path("model_args.d_vector_file", output_d_vector_file_path, config_path) + self._update_path("model_args.d_vector_file", output_d_vector_file_pth_path, config_path) # update the speaker_ids.json file path in the model config.json to the current path self._update_path("speakers_file", output_speaker_ids_file_path, config_path) + self._update_path("speakers_file", output_speaker_ids_file_pth_path, config_path) self._update_path("model_args.speakers_file", output_speaker_ids_file_path, config_path) + self._update_path("model_args.speakers_file", output_speaker_ids_file_pth_path, config_path) # update the speaker_encoder file path in the model config.json to the current path self._update_path("speaker_encoder_model_path", speaker_encoder_model_path, config_path)