From 0af00513994f3fb4ef96a6b846d3e4a88db241ba Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 20 Apr 2022 00:01:57 -0600 Subject: [PATCH] fix --- codes/models/audio/music/unet_diffusion_waveform_gen2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/models/audio/music/unet_diffusion_waveform_gen2.py b/codes/models/audio/music/unet_diffusion_waveform_gen2.py index 4c250074..8e61a737 100644 --- a/codes/models/audio/music/unet_diffusion_waveform_gen2.py +++ b/codes/models/audio/music/unet_diffusion_waveform_gen2.py @@ -140,6 +140,7 @@ class ResBlockSimple(nn.Module): def _forward(self, x): h = self.in_layers(x) + h = self.out_layers(h) return self.skip_connection(x) + h