deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

one more

Browse Source
This commit is contained in:
James Betker 2022-05-30 16:33:49 -06:00
parent 71cf654957
commit 29b55d42a5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/audio/music/transformer_diffusion5.py
View File

@ -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