Update config.json

config.json

@@ -1,12 +1,14 @@
 {
-  "_class_name": "ConsistencyDecoderVae",
+  "_class_name": "ConsistencyDecoderVAE",
   "_diffusers_version": "0.23.0.dev0",
+  "_name_or_path": "openai/consistency-decoder",
   "block_out_channels": [
     128,
     256,
     512,
     512
   ],
+  "decoder_add_attention": false,
   "decoder_args": {
     "_use_default_values": [
       "downsample_type",
@@ -66,6 +68,34 @@
     ],
     "upsample_type": "conv"
   },
+  "decoder_block_out_channels": [
+    320,
+    640,
+    1024,
+    1024
+  ],
+  "decoder_down_block_types": [
+    "ResnetDownsampleBlock2D",
+    "ResnetDownsampleBlock2D",
+    "ResnetDownsampleBlock2D",
+    "ResnetDownsampleBlock2D"
+  ],
+  "decoder_downsample_padding": 1,
+  "decoder_in_channels": 7,
+  "decoder_layers_per_block": 3,
+  "decoder_norm_eps": 1e-05,
+  "decoder_norm_num_groups": 32,
+  "decoder_num_train_timesteps": 1024,
+  "decoder_out_channels": 6,
+  "decoder_resnet_time_scale_shift": "scale_shift",
+  "decoder_time_embedding_type": "learned",
+  "decoder_up_block_types": [
+    "ResnetUpsampleBlock2D",
+    "ResnetUpsampleBlock2D",
+    "ResnetUpsampleBlock2D",
+    "ResnetUpsampleBlock2D"
+  ],
+  "encoder_act_fn": "silu",
   "encoder_args": {
     "act_fn": "silu",
     "block_out_channels": [
@@ -86,6 +116,23 @@
     "norm_num_groups": 32,
     "out_channels": 4
   },
+  "encoder_block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "encoder_double_z": true,
+  "encoder_down_block_types": [
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
+  ],
+  "encoder_in_channels": 3,
+  "encoder_layers_per_block": 2,
+  "encoder_norm_num_groups": 32,
+  "encoder_out_channels": 4,
   "latent_channels": 4,
   "scaling_factor": 0.18215
-}
+}