Create config.json

unet/config.json

@@ -0,0 +1,26 @@
+{
+  "_class_name": "SPADUnetModel",
+  "_diffusers_version": "0.25.0",
+  "attention_resolutions": [
+    4,
+    2,
+    1
+  ],
+  "channel_mult": [
+    1,
+    2,
+    4,
+    4
+  ],
+  "context_dim": 768,
+  "image_size": 32,
+  "in_channels": 4,
+  "model_channels": 320,
+  "num_heads": 8,
+  "num_res_blocks": 2,
+  "out_channels": 4,
+  "transformer_depth": 1,
+  "use_checkpoint": false,
+  "use_spatial_transformer": true,
+  "legacy": false
+}