diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py
index 28b8f01..6d20a91 100644
--- a/configs/vae_magvit_v2/inference/16x128x128.py
+++ b/configs/vae_magvit_v2/inference/16x128x128.py
@@ -33,7 +33,8 @@ model = dict(
     kl_embed_dim = 64,
     activation_fn = 'swish',
     separate_first_frame_encoding = False,
-    custom_conv_padding = None
+    custom_conv_padding = None,
+    encoder_double_z = False,
 )
 
 
diff --git a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
index 24ad8f6..540d63d 100644
--- a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
+++ b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
@@ -42,7 +42,8 @@ model = dict(
     activation_fn = 'swish',
     separate_first_frame_encoding = False,
     disable_space = True,
-    custom_conv_padding = None
+    custom_conv_padding = None,
+    encoder_double_z = False,
 )
 
 
diff --git a/configs/vae_magvit_v2/train/16x128x128_trial.py b/configs/vae_magvit_v2/train/16x128x128_trial.py
index 9de6e7f..7cd5d80 100644
--- a/configs/vae_magvit_v2/train/16x128x128_trial.py
+++ b/configs/vae_magvit_v2/train/16x128x128_trial.py
@@ -31,6 +31,7 @@ model = dict(
     activation_fn = 'swish',
     separate_first_frame_encoding = False,
     disable_space = False,
+    encoder_double_z = False,
     custom_conv_padding = None
 )
 
diff --git a/configs/vae_magvit_v2/train/pipeline_16x128x128.py b/configs/vae_magvit_v2/train/pipeline_16x128x128.py
index cab30fc..6da9ee1 100644
--- a/configs/vae_magvit_v2/train/pipeline_16x128x128.py
+++ b/configs/vae_magvit_v2/train/pipeline_16x128x128.py
@@ -36,6 +36,7 @@ model = dict(
     activation_fn = 'swish',
     separate_first_frame_encoding = False,
     disable_space = True,
+    ncoder_double_z = False,
     custom_conv_padding = None
 )