Merge pull request #742 from fijipants/patch-3

Fix GPU inference
2021-08-26 22:11:03 +02:00 · 2021-08-26 22:11:03 +02:00 · 8f196fb01d
parent b02c4fe347 8f57f8adfd
commit 8f196fb01d
1 changed files with 1 additions and 1 deletions
--- a/TTS/utils/synthesizer.py
+++ b/TTS/utils/synthesizer.py
@ -251,7 +251,7 @@ class Synthesizer(object):
                d_vector=speaker_embedding,
            )
            waveform = outputs["wav"]
-            mel_postnet_spec = outputs["outputs"]["model_outputs"][0].detach().numpy()
+            mel_postnet_spec = outputs["outputs"]["model_outputs"][0].detach().cpu().numpy()
            if not use_gl:
                # denormalize tts output based on tts audio config
                mel_postnet_spec = self.ap.denormalize(mel_postnet_spec.T).T