Fix #3787

coqui-ai · Jun 12, 2024 · 356eb70 · 356eb70
1 parent dbf1a08
commit 356eb70
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TTS/vc/models/freevc.py b/TTS/vc/models/freevc.py
@@ -379,7 +379,8 @@ def load_pretrained_speaker_encoder(self):
         """Load pretrained speaker encoder model as mentioned in the paper."""
         print(" > 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):