Upload config.json

unet/config.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "UNet3DConditionModel",
+  "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.6.0",
   "act_fn": "silu",
   "attention_head_dim": 8,
@@ -12,10 +12,10 @@
   "center_input_sample": false,
   "cross_attention_dim": 768,
   "down_block_types": [
-    "CrossAttnDownBlock3D",
-    "CrossAttnDownBlock3D",
-    "CrossAttnDownBlock3D",
-    "DownBlock3D"
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
   ],
   "downsample_padding": 1,
   "flip_sin_to_cos": true,
@@ -28,9 +28,9 @@
   "out_channels": 4,
   "sample_size": 64,
   "up_block_types": [
-    "UpBlock3D",
-    "CrossAttnUpBlock3D",
-    "CrossAttnUpBlock3D",
-    "CrossAttnUpBlock3D"
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D"
   ]
 }