blakelemoine/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

fix

Browse Source
This commit is contained in:
James Betker 2022-04-20 00:01:57 -06:00
parent 419f4d37bd
commit 0af0051399
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/audio/music/unet_diffusion_waveform_gen2.py
View File

@ -140,6 +140,7 @@ class ResBlockSimple(nn.Module):
def _forward(self, x): def _forward(self, x):
h = self.in_layers(x) h = self.in_layers(x)
h = self.out_layers(h)
return self.skip_connection(x) + h return self.skip_connection(x) + h