one more

2022-05-30 16:33:49 -06:00 · 2022-05-30 16:33:49 -06:00 · 29b55d42a5
commit 29b55d42a5
parent 71cf654957
1 changed files with 1 additions and 0 deletions
--- a/codes/models/audio/music/transformer_diffusion5.py
+++ b/codes/models/audio/music/transformer_diffusion5.py
@ -218,6 +218,7 @@ class TransformerDiffusionWithQuantizer(nn.Module):
        del self.m2v.m2v.projector.projection
        del self.m2v.project_hid
        del self.m2v.project_q
+        del self.m2v.m2v.masked_spec_embed

    def update_for_step(self, step, *args):
        self.internal_step = step