wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix multiband melgan inference

Browse Source
This commit is contained in:
erogol
2020-06-15 19:04:05 +02:00
parent c866f23af6
commit 1dfafe003d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vocoder/models/multiband_melgan_generator.py
+2 -1
View File
@@ -29,10 +29,11 @@ class MultibandMelganGenerator(MelganGenerator):
def pqmf_synthesis(self, x):
return self.pqmf_layer.synthesis(x)
@torch.no_grad()
def inference(self, cond_features):
cond_features = cond_features.to(self.layers[1].weight.device)
cond_features = torch.nn.functional.pad(
cond_features,
(self.inference_padding, self.inference_padding),
'replicate')
return self.pqmf.synthesis(self.layers(cond_features))
return self.pqmf_synthesis(self.layers(cond_features))