From 29b55d42a5cf261af9cefa4846522f7508741bde Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Mon, 30 May 2022 16:33:49 -0600
Subject: [PATCH] one more
---
codes/models/audio/music/transformer_diffusion5.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/codes/models/audio/music/transformer_diffusion5.py b/codes/models/audio/music/transformer_diffusion5.py
index fc938086..e9ef9a23 100644
--- 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