From c02c06b778497156ec58814e1734c51f977b89e0 Mon Sep 17 00:00:00 2001
From: rromb <robin.rombach@gmail.com>
Date: Fri, 10 Jun 2022 12:03:44 +0200
Subject: [PATCH] update ae config

---
 .../txt2img-multinode-clip-encoder-f16-256-pretraining.yaml      | 1 +
 1 file changed, 1 insertion(+)

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