From da367da411a8f3363960414374ac13401ae54a17 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 28 May 2022 22:30:23 -0600 Subject: [PATCH] df5 --- codes/models/audio/music/transformer_diffusion5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion5.py b/codes/models/audio/music/transformer_diffusion5.py index 77723c09..2ae84345 100644 --- a/codes/models/audio/music/transformer_diffusion5.py +++ b/codes/models/audio/music/transformer_diffusion5.py @@ -196,7 +196,7 @@ class TransformerDiffusion(nn.Module): @register_model -def register_transformer_diffusion4(opt_net, opt): +def register_transformer_diffusion5(opt_net, opt): return TransformerDiffusion(**opt_net['kwargs'])