Bug fix in speaker encoder load in inference time

2022-04-21 15:55:49 -03:00 · 2022-04-21 15:55:49 -03:00 · c32082a62c
parent adcc2f8299
commit c32082a62c
1 changed files with 1 additions and 0 deletions
--- a/TTS/utils/synthesizer.py
+++ b/TTS/utils/synthesizer.py
@ -122,6 +122,7 @@ class Synthesizer(object):
            self.tts_model.cuda()

        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)

    def _set_speaker_encoder_paths_from_tts_config(self):