fix: check if waveform is pt tensor before cpu conversion

coqui-ai · Aug 13, 2023 · ef554f9 · ef554f9
1 parent 41de849
commit ef554f9
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 waveform.device != torch.device("cpu") and not use_gl:
+                if torch.is_tensor(waveform) and 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 waveform.device != torch.device("cpu"):
+            if torch.is_tensor(waveform) and waveform.device != torch.device("cpu"):
                 waveform = waveform.cpu()
             if not use_gl:
                 waveform = waveform.numpy()