| { |
| "model_type": "petal_dit", |
| "name": "Anthos", |
| "architecture": { |
| "class": "DiT", |
| "input_size": 32, |
| "patch_size": 2, |
| "in_channels": 4, |
| "hidden_size": 96, |
| "depth": 6, |
| "num_heads": 4, |
| "mlp_ratio": 2.0, |
| "learn_sigma": false |
| }, |
| "conditioning": { |
| "num_classes": 102, |
| "class_dropout_prob": 0.1 |
| }, |
| "vae": { |
| "name": "stabilityai/sd-vae-ft-ema", |
| "scale_factor": 0.18215 |
| }, |
| "sampling": { |
| "default_steps": 50, |
| "default_cfg_scale": 4.0, |
| "default_sampler": "heun" |
| }, |
| "weights": "model.safetensors", |
| "class_names_file": "classes.txt" |
| } |
|
|
