fixing torch device error when synthesizer is called with use_cuda=True

coqui-ai · Dec 20, 2023 · 6900633 · 6900633
1 parent 55c7063
commit 6900633
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/tts/models/forward_tts.py b/TTS/tts/models/forward_tts.py
@@ -398,7 +398,7 @@ def _forward_encoder(
         """
         if hasattr(self, "emb_g"):
             g = g.type(torch.LongTensor)
-            g = self.emb_g(g)  # [B, C, 1]
+            g = self.emb_g(g.to("cuda") if torch.cuda.is_available() else g.to("cpu"))  # [B, C, 1]
         if g is not None:
             g = g.unsqueeze(-1)
         # [B, T, C]