diff --git a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml
index cacf46e..19b498f 100644
--- a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml
+++ b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml
@@ -49,6 +49,7 @@ model:
         monitor: val/rec_loss
         ckpt_path: "models/first_stage_models/kl-f16/model.ckpt"
         ddconfig:
+          double_z: True
           z_channels: 16
           resolution: 256
           in_channels: 3