Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rename set_embeddings_from_file to load_embeddings_from_file

This commit is contained in:
Edresson Casanova 2022-03-14 13:34:16 +00:00
parent 7e59755d63
commit 464775dbaf
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/utils/managers.py
View File

@ -99,7 +99,7 @@ class EmbeddingManager(BaseIDManager):
self.use_cuda = use_cuda self.use_cuda = use_cuda
if embedding_file_path: 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: if encoder_model_path and encoder_config_path:
self.init_encoder(encoder_model_path, 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) 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. """Load embeddings from a json file.
Args: Args:

4
TTS/tts/utils/speakers.py
View File

@ -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" "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.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. elif not c.use_d_vector_file: # restor speaker manager with speaker ID file.
speaker_ids_from_data = speaker_manager.ids speaker_ids_from_data = speaker_manager.ids
speaker_manager.load_ids_from_file(speakers_file) 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." ), " [!] You cannot introduce new speakers to a pre-trained model."
elif c.use_d_vector_file and c.d_vector_file: elif c.use_d_vector_file and c.d_vector_file:
# new speaker manager with external speaker embeddings. # 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: 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." 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: elif c.use_speaker_embedding and "speakers_file" in c and c.speakers_file: