Update config.json

config.json

@@ -3,6 +3,11 @@
   "attn_resolutions": [
     16
   ],
+  "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"],
+  "up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"],
+  "conv_resample": true,
+  "downsample_padding": 0,
+  "num_head_channels": null,
   "ch": 128,
   "ch_mult": [
     1,
@@ -12,11 +17,18 @@
     4,
     4
   ],
+  "block_channels": [128, 128, 256, 256, 512, 512],
+  "resnet_eps": 1e-6,
+  "flip_sin_to_cos": false,
+  "downscale_freq_shift": 1,
   "dropout": 0.0,
   "in_channels": 3,
   "name_or_path": "./ddpm-lsun-church/",
   "num_res_blocks": 2,
   "out_ch": 3,
+  "out_channels": 3,
   "resamp_with_conv": true,
-  "resolution": 256
+  "resolution": 256,
+  "image_size": 32,
+  "ddpm": true
 }