config.json

{
  "architectures": ["DigitDiffusionModel"],
  "auto_map": {
    "AutoConfig": "configuration.DigitDiffusionConfig",
    "AutoModel": "modeling.DigitDiffusionModel"
  },
  "model_type": "digit_diffusion",
  "image_size": 32,
  "sample_size": 32,
  "in_channels": 1,
  "out_channels": 1,
  "num_classes": 10,
  "block_out_channels": [12, 16],
  "layers_per_block": 8,
  "norm_num_groups": 4,
  "cross_attention_dim": 8,
  "class_embed_type": null
}