From aca9024d9b2ee8ad6a16ea6756e6fa7af2fcc6a0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 10 Jun 2022 16:23:08 -0600 Subject: [PATCH] qcodes --- 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 98c9ab21..973c83ca 100644 --- a/codes/models/audio/music/transformer_diffusion8.py +++ b/codes/models/audio/music/transformer_diffusion8.py @@ -237,7 +237,7 @@ class TransformerDiffusionWithQuantizer(nn.Module): def get_debug_values(self, step, __): if self.quantizer.total_codes > 0: - return {'histogram_codes': self.quantizer.codes[:self.quantizer.total_codes], + return {'histogram_quant_codes': self.quantizer.codes[:self.quantizer.total_codes], 'gumbel_temperature': self.quantizer.quantizer.temperature} else: return {}