Update synthesizer.py

2021-08-18 19:56:52 -04:00 · 2021-08-18 19:56:52 -04:00 · 8f57f8adfd
parent b02c4fe347
commit 8f57f8adfd
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