| { |
| "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"] |
| } |
|
|
