fix(freevc): use the specified device for pretrained speaker encoder (#…

…45) Fixes coqui-ai#3787
idiap · Jun 16, 2024 · 3a20f47 · 3a20f47
1 parent 063e9e9
commit 3a20f47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TTS/vc/models/freevc.py b/TTS/vc/models/freevc.py
@@ -382,7 +382,8 @@ def load_pretrained_speaker_encoder(self):
         """Load pretrained speaker encoder model as mentioned in the paper."""
         logger.info("Loading pretrained speaker encoder model ...")
         self.enc_spk_ex = SpeakerEncoderEx(
-            "https://github.com/coqui-ai/TTS/releases/download/v0.13.0_models/speaker_encoder.pt"
+            "https://github.com/coqui-ai/TTS/releases/download/v0.13.0_models/speaker_encoder.pt",
+            device=self.device
         )
 
     def init_multispeaker(self, config: Coqpit):