wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-25 13:20:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Init speaker manager for speaker encoder

Browse Source
This commit is contained in:
Eren Gölge
2021-12-22 15:51:53 +00:00
parent 633dcc9c56
commit 5c5ddd2ba7
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/utils/synthesizer.py
+8 -1
View File
@@ -171,7 +171,12 @@ class Synthesizer(object):
def _init_speaker_encoder(self, speaker_manager):
"""Initialize the SpeakerEncoder"""
if self.encoder_checkpoint:
speaker_manager.init_speaker_encoder(self.encoder_checkpoint, self.encoder_config)
if speaker_manager is None:
speaker_manager = SpeakerManager(
encoder_model_path=self.encoder_checkpoint, encoder_config_path=self.encoder_config
)
else:
speaker_manager.init_speaker_encoder(self.encoder_checkpoint, self.encoder_config)
return speaker_manager
def _init_language_manager(self):
@@ -183,6 +188,8 @@ class Synthesizer(object):
language_manager = LanguageManager(language_ids_file_path=self.tts_languages_file)
elif self.tts_config.get("language_ids_file", None):
language_manager = LanguageManager(language_ids_file_path=self.tts_config.language_ids_file)
else:
language_manager = LanguageManager(config=self.tts_config)
return language_manager
def _load_vocoder(self, model_file: str, model_config: str, use_cuda: bool) -> None: