mirror of https://github.com/coqui-ai/TTS.git
Merge pull request #353 from richardburleigh/patch-1
Fix GL overriding PWGAN inference
This commit is contained in:
commit
a9632d56d8
|
@ -172,7 +172,7 @@ class Synthesizer(object):
|
||||||
if self.use_cuda:
|
if self.use_cuda:
|
||||||
vocoder_input.cuda()
|
vocoder_input.cuda()
|
||||||
wav = self.pwgan.inference(vocoder_input, hop_size=self.ap.hop_length)
|
wav = self.pwgan.inference(vocoder_input, hop_size=self.ap.hop_length)
|
||||||
if self.wavernn:
|
elif self.wavernn:
|
||||||
vocoder_input = torch.FloatTensor(postnet_output.T).unsqueeze(0)
|
vocoder_input = torch.FloatTensor(postnet_output.T).unsqueeze(0)
|
||||||
if self.use_cuda:
|
if self.use_cuda:
|
||||||
vocoder_input.cuda()
|
vocoder_input.cuda()
|
||||||
|
|
Loading…
Reference in New Issue