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

harharhack

Browse Source
This commit is contained in:
James Betker 2022-06-10 15:13:24 -06:00
parent 7198bd8bd0
commit 33178e89c4
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_diffusion8.py
View File

@ -228,7 +228,7 @@ class TransformerDiffusionWithQuantizer(nn.Module):
for p in self.diff.parameters():
unused = unused + p.mean() * 0
mse = mse + unused
return x, diversity_loss, mse
return x.repeat(1,2,1), diversity_loss, mse
quant_grad_enabled = self.internal_step >= self.freeze_quantizer_until
with torch.set_grad_enabled(quant_grad_enabled):