fix

2022-04-20 00:01:57 -06:00 · 2022-04-20 00:01:57 -06:00 · 0af0051399
commit 0af0051399
parent 419f4d37bd
1 changed files with 1 additions and 0 deletions
--- 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