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