Update vae/config.json

vae/config.json

@@ -10,10 +10,10 @@
     512
   ],
   "down_block_types": [
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D",
-    "DownEncoderBlock2D"
+    "DownEncoderBlock3D",
+    "DownEncoderBlock3D",
+    "DownEncoderBlock3D",
+    "DownEncoderBlock3D"
   ],
   "extra_cond_dim": null,
   "force_upcast": true,
@@ -26,20 +26,14 @@
   "latents_mean": null,
   "latents_std": null,
   "layers_per_block": 2,
-  "memory_limit": {
-    "conv_max_mem": 0.5,
-    "norm_max_mem": 0.5
-  },
+  "memory_limit": null,
   "mid_block_add_attention": true,
   "norm_num_groups": 32,
   "out_channels": 3,
   "sample_size": 32,
   "scaling_factor": 0.9152,
   "shift_factor": null,
-  "slicing": {
-    "memory_device": "same",
-    "split_size": 4
-  },
+  "slicing": null,
   "slicing_sample_min_size": 4,
   "slicing_up_num": 0,
   "spatial_downsample_factor": 8,
@@ -47,11 +41,11 @@
   "temporal_scale_num": 2,
   "time_receptive_field": "full",
   "up_block_types": [
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D",
-    "UpDecoderBlock2D"
+    "UpDecoderBlock3D",
+    "UpDecoderBlock3D",
+    "UpDecoderBlock3D",
+    "UpDecoderBlock3D"
   ],
-  "use_post_quant_conv": true,
-  "use_quant_conv": true
+  "use_post_quant_conv": false,
+  "use_quant_conv": false
 }