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