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_ids_from_file to load_ids_from_file

This commit is contained in:
Edresson Casanova 2022-03-14 13:31:01 +00:00
parent 25da4d9b74
commit 7e59755d63
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

@ -22,7 +22,7 @@ class BaseIDManager:
self.ids = {}
if id_file_path:
self.set_ids_from_file(id_file_path)
self.load_ids_from_file(id_file_path)
@staticmethod
def _load_json(json_file_path: str) -> Dict:
@ -43,7 +43,7 @@ class BaseIDManager:
"""
self.ids, _ = self.parse_ids_from_data(items)
def set_ids_from_file(self, file_path: str) -> None:
def load_ids_from_file(self, file_path: str) -> None:
"""Set IDs from a file.
Args:

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

@ -196,7 +196,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None,
speaker_manager.set_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.set_ids_from_file(speakers_file)
speaker_manager.load_ids_from_file(speakers_file)
assert all(
speaker in speaker_manager.ids for speaker in speaker_ids_from_data
), " [!] You cannot introduce new speakers to a pre-trained model."
@ -207,7 +207,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None,
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:
# new speaker manager with speaker IDs file.
speaker_manager.set_ids_from_file(c.speakers_file)
speaker_manager.load_ids_from_file(c.speakers_file)
if speaker_manager.num_speakers > 0:
print(