From 5028703b3d1e3de40c9b894b2c4bad24b61119c7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 6 Jun 2022 09:26:25 -0600 Subject: [PATCH] ci not required --- codes/models/audio/music/transformer_diffusion9.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion9.py b/codes/models/audio/music/transformer_diffusion9.py index e9c30ef6..6670b7f3 100644 --- a/codes/models/audio/music/transformer_diffusion9.py +++ b/codes/models/audio/music/transformer_diffusion9.py @@ -253,7 +253,7 @@ class TransformerDiffusionWithQuantizer(nn.Module): self.quantizer.min_gumbel_temperature, ) - def forward(self, x, timesteps, truth_mel, conditioning_input, disable_diversity=False, conditioning_free=False): + def forward(self, x, timesteps, truth_mel, conditioning_input=None, disable_diversity=False, conditioning_free=False): quant_grad_enabled = self.internal_step > self.freeze_quantizer_until with torch.set_grad_enabled(quant_grad_enabled): proj, diversity_loss = self.quantizer(truth_mel, return_decoder_latent=True)