mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update params

Browse Source
This commit is contained in:
James Betker 2022-06-19 20:47:06 -06:00
parent ff8b0533ac
commit 02ead8c05c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/unet_diffusion_waveform_gen3.py
View File

@ -312,7 +312,7 @@ class DiffusionWaveformGen(nn.Module):
groups = { groups = {
'input_blocks': list(self.input_blocks.parameters()), 'input_blocks': list(self.input_blocks.parameters()),
'output_blocks': list(self.output_blocks.parameters()), 'output_blocks': list(self.output_blocks.parameters()),
'middle_transformer': list(self.middle_block.parameters()), 'middle_rrdb': list(self.middle_block.parameters()),
} }
return groups return groups