config.json

{
  "architectures": ["DigitDiffusionModel"],
  "auto_map": {
    "AutoConfig": "configuration.DigitDiffusionConfig",
    "AutoModel": "modeling.DigitDiffusionModel"
  },
  "block_out_channels": [12, 16, 20],
  "class_embed_type": null,
  "cross_attention_dim": 8,
  "down_block_types": ["DownBlock2D", "DownBlock2D", "DownBlock2D"],
  "image_size": 32,
  "in_channels": 1,
  "layers_per_block": 8,
  "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"]
}