mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

whoops!

Browse Source
This commit is contained in:
James Betker 2022-06-15 09:01:20 -06:00
parent ff5c03b460
commit b51ff8a176
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/transformer_diffusion12.py
View File

@ -168,7 +168,7 @@ class TransformerDiffusion(nn.Module):
for p in self.parameters(): for p in self.parameters():
p.DO_NOT_TRAIN = True p.DO_NOT_TRAIN = True
p.requires_grad = False p.requires_grad = False
for m in [self.input_converter and self.code_converter]: for m in [self.ar_input and self.ar_prior_intg]:
for p in m.parameters(): for p in m.parameters():
del p.DO_NOT_TRAIN del p.DO_NOT_TRAIN
p.requires_grad = True p.requires_grad = True