fix: add necessary vocoder output device check

coqui-ai · Aug 10, 2023 · 0fa95a2 · 0fa95a2
1 parent 586e5be
commit 0fa95a2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/TTS/utils/synthesizer.py b/TTS/utils/synthesizer.py
@@ -410,7 +410,7 @@ def tts(
                     # run vocoder model
                     # [1, T, C]
                     waveform = self.vocoder_model.inference(vocoder_input.to(vocoder_device))
-                if not use_gl and waveform.device != torch.device("cpu"):
+                if waveform.device != torch.device("cpu") and not use_gl:
                     waveform = waveform.cpu()
                 if not use_gl:
                     waveform = waveform.numpy()
@@ -474,7 +474,7 @@ def tts(
                 # run vocoder model
                 # [1, T, C]
                 waveform = self.vocoder_model.inference(vocoder_input.to(vocoder_device))
-            if not use_gl and waveform.device != torch.device("cpu"):
+            if waveform.device != torch.device("cpu"):
                     waveform = waveform.cpu()
             if not use_gl:
                 waveform = waveform.numpy()