Update config.json

config.json

@@ -3,6 +3,11 @@
   "attn_resolutions": [
     16
   ],
+  "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"],
+  "up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"],
+  "conv_resample": true,
+  "downsample_padding": 0,
+  "num_head_channels": 64,
   "ch": 128,
   "ch_mult": [
     1,
@@ -10,11 +15,17 @@
     2,
     2
   ],
+  "block_channels": [128, 256, 256, 256],
+  "resnet_eps": 1e-6,
+  "flip_sin_to_cos": false,
+  "downscale_freq_shift": 1,
   "dropout": 0.1,
   "in_channels": 3,
   "name_or_path": "./ddpm-cifar10/",
   "num_res_blocks": 2,
   "out_ch": 3,
   "resamp_with_conv": true,
-  "resolution": 32
+  "resolution": 32,
+  "image_size": 32,
+  "ddpm": true
 }