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