diff --git a/codes/models/audio/music/transformer_diffusion12.py b/codes/models/audio/music/transformer_diffusion12.py
index bd5cd0b5..f82a267d 100644
--- a/codes/models/audio/music/transformer_diffusion12.py
+++ b/codes/models/audio/music/transformer_diffusion12.py
@@ -407,7 +407,7 @@ class TransformerDiffusionWithPretrainedVqvae(nn.Module):
 
 
 class TransformerDiffusionWithMultiPretrainedVqvae(nn.Module):
-    def __init__(self, num_vaes=4, vqargs, **kwargs):
+    def __init__(self, num_vaes=4, vqargs={}, **kwargs):
         super().__init__()
 
         self.internal_step = 0