From 33178e89c4194539f78e1ea4e67c777260c66611 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 10 Jun 2022 15:13:24 -0600 Subject: [PATCH] harharhack --- codes/models/audio/music/transformer_diffusion8.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion8.py b/codes/models/audio/music/transformer_diffusion8.py index 145a7260..0037ec5c 100644 --- a/codes/models/audio/music/transformer_diffusion8.py +++ b/codes/models/audio/music/transformer_diffusion8.py @@ -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):