From 8552c1d991955c8545e59b7d1af492449ee82c9a Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 9 Jun 2020 23:02:55 +0200 Subject: [PATCH] fix multiband melgan inference --- vocoder/models/multiband_melgan_generator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vocoder/models/multiband_melgan_generator.py b/vocoder/models/multiband_melgan_generator.py index 8feacd25..15e7426e 100644 --- a/vocoder/models/multiband_melgan_generator.py +++ b/vocoder/models/multiband_melgan_generator.py @@ -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))