config.json

{
  "model_type": "agate",
  "arch": "fcdm_t2",
  "model_kw": {
    "widths": [
      288,
      832,
      1664
    ],
    "depths": [
      4,
      6,
      2
    ],
    "steer_rank": [
      128,
      256,
      256
    ],
    "ctx_dim": 512,
    "dim": 640,
    "prelude": 3,
    "core": 3,
    "coda": 2,
    "loops": 4,
    "loop_probs": [
      0.0,
      0.0,
      0.0,
      1.0
    ]
  },
  "params_generator": 190929092,
  "text_encoder": "text_encoder",
  "text_max_len": 512,
  "guide": {
    "generator": "guide/generator.safetensors",
    "text_encoder": "guide/text_encoder",
    "text_max_len": 128,
    "step": 27600
  },
  "vae": "stabilityai/sd-vae-ft-mse",
  "vae_scale": 0.18215,
  "fast_vae": "madebyollin/taesd",
  "resolution": 256,
  "latent_hw": 32,
  "sampler": {
    "kind": "euler",
    "steps": 50,
    "cfg": 3.0,
    "t0_is_noise": true,
    "uncond_prompt": ""
  },
  "training": {
    "step": 99250,
    "samples_seen": 146900000,
    "gpu_hours_gh200": 144.7,
    "energy_kwh_at_wall": 81.8,
    "co2e_kg": 2.45,
    "dataset": "LucasFang/FLUX-Reason-6M"
  }
}