malarinv
/
tacotron2
mirror of https://github.com/malarinv/tacotron2
1
0
Fork 0
Code Issues Releases Wiki Activity

bugfix for vocoder arg

master
Malar 2019-11-28 17:03:46 +05:30
parent 009b87e716
commit 78eed2d295
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
taco2/tts.py
View File

@ -104,12 +104,12 @@ class TTSModel(object):
def synth_speech_array(self, text, vocoder): def synth_speech_array(self, text, vocoder):
mel_outputs_postnet = self.generate_mel_postnet(text) mel_outputs_postnet = self.generate_mel_postnet(text)
if method == "wavglow": if vocoder == "wavglow":
with torch.no_grad(): with torch.no_grad():
audio_t = self.waveglow.infer(mel_outputs_postnet, sigma=0.666) audio_t = self.waveglow.infer(mel_outputs_postnet, sigma=0.666)
audio_t = self.denoiser(audio_t, 0.1)[0] audio_t = self.denoiser(audio_t, 0.1)[0]
audio = audio_t[0].data.cpu().numpy() audio = audio_t[0].data.cpu().numpy()
elif method == "gl": elif vocoder == "gl":
mel_decompress = self.taco_stft.spectral_de_normalize(mel_outputs_postnet) mel_decompress = self.taco_stft.spectral_de_normalize(mel_outputs_postnet)
mel_decompress = mel_decompress.transpose(1, 2).data.cpu() mel_decompress = mel_decompress.transpose(1, 2).data.cpu()
spec_from_mel_scaling = 1000 spec_from_mel_scaling = 1000