.

2022-06-17 09:40:11 +07:00 · 2022-06-17 09:40:11 +07:00 · c000e489fa
parent 7ca532c7cc
commit c000e489fa
1 changed files with 1 additions and 0 deletions
--- a/codes/models/audio/music/transformer_diffusion12.py
+++ b/codes/models/audio/music/transformer_diffusion12.py
@ -540,6 +540,7 @@ class TransformerDiffusionWithCheaterLatent(nn.Module):

    def get_debug_values(self, step, __):
        self.internal_step = step
+        return {}

    def get_grad_norm_parameter_groups(self):
        attn1 = list(itertools.chain.from_iterable([lyr.block1.attn.parameters() for lyr in self.diff.layers]))