Merge pull request #742 from fijipants/patch-3

Fix GPU inference
This commit is contained in:
Eren Gölge
2021-08-26 22:11:03 +02:00
committed by GitHub

View File

@@ -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