From 02ead8c05c0e9ae39d30f74395c960e3c165aac4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 19 Jun 2022 20:47:06 -0600 Subject: [PATCH] update params --- codes/models/audio/music/unet_diffusion_waveform_gen3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/unet_diffusion_waveform_gen3.py b/codes/models/audio/music/unet_diffusion_waveform_gen3.py index 857a849e..3610f882 100644 --- a/codes/models/audio/music/unet_diffusion_waveform_gen3.py +++ b/codes/models/audio/music/unet_diffusion_waveform_gen3.py @@ -312,7 +312,7 @@ class DiffusionWaveformGen(nn.Module): groups = { 'input_blocks': list(self.input_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