diff --git a/configs/vae/train/video.py b/configs/vae/train/video.py
index dda530d..02bbb24 100644
--- a/configs/vae/train/video.py
+++ b/configs/vae/train/video.py
@@ -19,7 +19,7 @@ plugin = "zero2"
 # Define model
 model = dict(
     type="VideoAutoencoderPipeline",
-    freeze_vae_2d=True,
+    freeze_vae_2d=False,
     from_pretrained=None,
     vae_2d=dict(
         type="VideoAutoencoderKL",
@@ -39,7 +39,7 @@ kl_loss_weight = 1e-6
 
 mixed_image_ratio = 0.2
 use_real_rec_loss = True
-use_z_rec_loss = True
+use_z_rec_loss = False
 use_image_identity_loss = False
 
 # Others