From 464775dbaf2ec0437e0271a6286f797cd13d0540 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Mon, 14 Mar 2022 13:34:16 +0000 Subject: [PATCH] Rename set_embeddings_from_file to load_embeddings_from_file --- TTS/tts/utils/managers.py | 4 ++-- TTS/tts/utils/speakers.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/TTS/tts/utils/managers.py b/TTS/tts/utils/managers.py index 03650d15..e19e08a7 100644 --- a/TTS/tts/utils/managers.py +++ b/TTS/tts/utils/managers.py @@ -99,7 +99,7 @@ class EmbeddingManager(BaseIDManager): self.use_cuda = use_cuda if embedding_file_path: - self.set_embeddings_from_file(embedding_file_path) + self.load_embeddings_from_file(embedding_file_path) if encoder_model_path and encoder_config_path: self.init_encoder(encoder_model_path, encoder_config_path) @@ -119,7 +119,7 @@ class EmbeddingManager(BaseIDManager): """ self._save_json(file_path, self.embeddings) - def set_embeddings_from_file(self, file_path: str) -> None: + def load_embeddings_from_file(self, file_path: str) -> None: """Load embeddings from a json file. Args: diff --git a/TTS/tts/utils/speakers.py b/TTS/tts/utils/speakers.py index b02b7ccc..f3a2845f 100644 --- a/TTS/tts/utils/speakers.py +++ b/TTS/tts/utils/speakers.py @@ -193,7 +193,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None, "You must copy the file speakers.json to restore_path, or set a valid file in CONFIG.d_vector_file" ) speaker_manager.load_embeddings_file(c.d_vector_file) - speaker_manager.set_embeddings_from_file(speakers_file) + speaker_manager.load_embeddings_from_file(speakers_file) elif not c.use_d_vector_file: # restor speaker manager with speaker ID file. speaker_ids_from_data = speaker_manager.ids speaker_manager.load_ids_from_file(speakers_file) @@ -202,7 +202,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None, ), " [!] You cannot introduce new speakers to a pre-trained model." elif c.use_d_vector_file and c.d_vector_file: # new speaker manager with external speaker embeddings. - speaker_manager.set_embeddings_from_file(c.d_vector_file) + speaker_manager.load_embeddings_from_file(c.d_vector_file) elif c.use_d_vector_file and not c.d_vector_file: raise "use_d_vector_file is True, so you need pass a external speaker embedding file." elif c.use_speaker_embedding and "speakers_file" in c and c.speakers_file: