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

Update SpeakerManager init in Synthesizer

This commit is contained in:
Eren Gölge 2022-05-11 11:32:27 +02:00
parent c3f8c4d5eb
commit 2fc38f67d2
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/utils/synthesizer.py
View File

@ -97,10 +97,10 @@ class Synthesizer(object):
"""Load the TTS model. """Load the TTS model.
1. Load the model config. 1. Load the model config.
2. Init the AudioProcessor. 2. Init the model from the config.
3. Init the model from the config. 3. Load the model weights.
4. Move the model to the GPU if CUDA is enabled. 4. Move the model to the GPU if CUDA is enabled.
5. Init the speaker manager for the model. 5. Init the speaker manager in the model.
Args: Args:
tts_checkpoint (str): path to the model checkpoint. tts_checkpoint (str): path to the model checkpoint.
@ -122,8 +122,7 @@ class Synthesizer(object):
self.tts_model.cuda() self.tts_model.cuda()
if self.encoder_checkpoint and hasattr(self.tts_model, "speaker_manager"): if self.encoder_checkpoint and hasattr(self.tts_model, "speaker_manager"):
self.tts_model.speaker_manager.use_cuda = use_cuda self.tts_model.speaker_manager.init_encoder(self.encoder_checkpoint, self.encoder_config, use_cuda)
self.tts_model.speaker_manager.init_encoder(self.encoder_checkpoint, self.encoder_config)
def _set_speaker_encoder_paths_from_tts_config(self): def _set_speaker_encoder_paths_from_tts_config(self):
"""Set the encoder paths from the tts model config for models with speaker encoders.""" """Set the encoder paths from the tts model config for models with speaker encoders."""