Update vae/config.json

vae/config.json

@@ -1,36 +1,29 @@
 {
-  "_class_name": "UNet2DConditionModel",
+  "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.2.2",
   "act_fn": "silu",
-  "attention_head_dim": 8,
   "block_out_channels": [
-    320,
-    640,
-    1280,
-    1280
+    128,
+    256,
+    512,
+    512
   ],
-  "center_input_sample": false,
-  "cross_attention_dim": 768,
   "down_block_types": [
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
-    "DownBlock2D"
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
   ],
-  "downsample_padding": 1,
-  "flip_sin_to_cos": true,
-  "freq_shift": 0,
-  "in_channels": 4,
+  "in_channels": 6,
+  "latent_channels": 4,
   "layers_per_block": 2,
-  "mid_block_scale_factor": 1,
-  "norm_eps": 1e-05,
-  "norm_num_groups": 32,
-  "out_channels": 4,
-  "sample_size": 64,
+  "out_channels": 6,
+  "sample_size": 256,
+  "scaling_factor": 0.18215,
   "up_block_types": [
-    "UpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D"
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D"
   ]
 }