diff --git a/opensora/models/ae/videobase/vqvae/configuration_vqvae.py b/opensora/models/ae/videobase/vqvae/configuration_vqvae.py
index 450735aeb..90ac29cfa 100644
--- a/opensora/models/ae/videobase/vqvae/configuration_vqvae.py
+++ b/opensora/models/ae/videobase/vqvae/configuration_vqvae.py
@@ -29,3 +29,5 @@ class VQVAEConfiguration(VideoBaseConfiguration):
             self.downsample = downsample
 
         self.no_pos_embd = no_pos_embd
+
+        self.hidden_size = n_hiddens