Update config.json

config.json

@@ -2,11 +2,13 @@
   "model_type": "UNet",
   "T": 1000,
   "channel": 128,
-  "channel_mult": [1, 2, 3, 4],
-  "attn": [2],
+  "channel_mult": [1, 2, 2, 2],
+  "attn": [1],
   "num_res_blocks": 2,
-  "dropout": 0.15,
+  "dropout": 0.1,
   "image_size": 32,
   "in_channels": 3,
-  "out_channels": 3
+  "out_channels": 3,
+  "time_embedding_dim": 512,  // 128 * 4
+  "norm_num_groups": 32
 }