Update vae/config.json

vae/config.json

@@ -1,12 +1,12 @@
 {
   "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.18.2",
+  "_diffusers_version": "0.24.0",
   "act_fn": "silu",
   "block_out_channels": [
-    128,
     128,
     256,
-    256
+    512,
+    512
   ],
   "down_block_types": [
     "DownEncoderBlock2D",
@@ -14,6 +14,7 @@
     "DownEncoderBlock2D",
     "DownEncoderBlock2D"
   ],
+  "force_upcast": true,
   "in_channels": 3,
   "latent_channels": 4,
   "layers_per_block": 2,
@@ -26,9 +27,5 @@
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",
     "UpDecoderBlock2D"
-  ],
-  "double_z": true,
-  "use_quant_conv": true,
-  "use_post_quant_conv": true,
-  "dropout": 0.0
-}
+  ]
+}