From 56c4a00e715be4724c3bf59d62d2c90922f589b3 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 19 Jun 2022 23:22:30 -0600 Subject: [PATCH] whoops --- .../music/transformer_diffusion_with_point_conditioning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion_with_point_conditioning.py b/codes/models/audio/music/transformer_diffusion_with_point_conditioning.py index 68b09b4a..444163fd 100644 --- a/codes/models/audio/music/transformer_diffusion_with_point_conditioning.py +++ b/codes/models/audio/music/transformer_diffusion_with_point_conditioning.py @@ -233,7 +233,7 @@ class TransformerDiffusionWithConditioningEncoder(nn.Module): def get_grad_norm_parameter_groups(self): groups = self.diff.get_grad_norm_parameter_groups() groups['conditioning_encoder'] = list(self.conditioning_encoder.parameters()) - return + return groups def before_step(self, step): scaled_grad_parameters = list(itertools.chain.from_iterable([lyr.out.parameters() for lyr in self.diff.layers])) + \