{
 "architecture": "TinyDiT",
 "latent_ch": 32,
 "ctx_dim": 768,
 "patch": 2,
 "n_registers": 16,
 "register_block": 3,
 "n_null": 2,
 "dim": 896,
 "depth": 16,
 "heads": 14,
 "autoencoder": "black-forest-labs/FLUX.2-dev (vae subfolder)",
 "text_encoder": "google/flan-t5-base",
 "max_tokens_long": 128,
 "max_tokens_short": 48,
 "latent_whitening": "stats.json (per-channel mean/std, applied after encode and undone before decode)",
 "sampler": {
  "steps": 20,
  "cfg": 4.0,
  "shift": 2.8,
  "solver": "euler"
 },
 "training_shapes": [
  [
   256,
   256
  ],
  [
   288,
   224
  ],
  [
   224,
   288
  ],
  [
   320,
   208
  ],
  [
   208,
   320
  ]
 ],
 "step": 400000,
 "ema": "0.9999, bf16"
}