Update config.json

config.json

@@ -4,19 +4,15 @@
     "AutoConfig": "configuration.DigitDiffusionConfig",
     "AutoModel": "modeling.DigitDiffusionModel"
   },
-  "block_out_channels": [12, 16],
-  "class_embed_type": null,
-  "cross_attention_dim": 8,
-  "down_block_types": ["DownBlock2D", "DownBlock2D", "DownBlock2D"],
+  "model_type": "digit_diffusion",
   "image_size": 32,
+  "sample_size": 32,
   "in_channels": 1,
+  "out_channels": 1,
+  "num_classes": 10,
+  "block_out_channels": [12, 16, 20],
   "layers_per_block": 5,
-  "mid_block_type": "UNetMidBlock2D",
-  "model_type": "digit_diffusion",
   "norm_num_groups": 4,
-  "num_class_embeds": 10,
-  "num_classes": 10,
-  "out_channels": 1,
-  "sample_size": 32,
-  "up_block_types": ["UpBlock2D", "UpBlock2D", "UpBlock2D"]
-}
+  "cross_attention_dim": 8,
+  "class_embed_type": null
+}