From c32082a62c101e619f919f45782fa6de4d9c72af Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Thu, 21 Apr 2022 15:55:49 -0300 Subject: [PATCH] Bug fix in speaker encoder load in inference time --- TTS/utils/synthesizer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/TTS/utils/synthesizer.py b/TTS/utils/synthesizer.py index 1a49f0b0..05161a66 100644 --- 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):