update ae config

2022-06-10 12:03:44 +02:00 · 2022-06-10 12:03:44 +02:00 · c02c06b778
parent dd10f44cdb
commit c02c06b778
1 changed files with 1 additions and 0 deletions
--- 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