From 08597bfaf520c04188c2dd506e53b04e7d2d1d2d Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 6 Jun 2022 09:21:58 -0600 Subject: [PATCH] fix --- 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 ce660866..e9c30ef6 100644 --- a/codes/models/audio/music/transformer_diffusion9.py +++ b/codes/models/audio/music/transformer_diffusion9.py @@ -299,7 +299,7 @@ def register_transformer_diffusion9(opt_net, opt): @register_model -def register_transformer_diffusion8_with_quantizer(opt_net, opt): +def register_transformer_diffusion9_with_quantizer(opt_net, opt): return TransformerDiffusionWithQuantizer(**opt_net['kwargs'])