{
  "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": 138010,
    "samples_seen": 186634240,
    "continued_from": "T40r caption phase, step 82,800 (the run that became Agate Preview 001)",
    "gpu_hours_gh200_lineage": 187.7,
    "energy_kwh_at_wall_lineage": 107.1,
    "co2e_kg_lineage": 3.21,
    "dataset": "LucasFang/FLUX-Reason-6M",
    "lineage_note": "whole lineage from scratch, jobs pro-rated by steps where this release branched mid-run (perun ledger, PUE 1.3, 30 g/kWh)"
  },
  "training_compute_flop": {
    "estimate": 6.5e+19,
    "range": [
      5.8e+19,
      7.2e+19
    ],
    "including_text_encoder_pretraining": 8.6e+20,
    "method": "measured forward FLOPs per image (torch.utils.flop_counter, batch 1, 128-token context) x images seen x 3 (forward + backward); Ettin at 128 tokens (low) to 4x (high); excludes the compute used by third parties to generate the synthetic training images",
    "forward_gflop_per_image": {
      "256": 94.4
    }
  },
  "output_marking": {
    "watermark": "invisible-watermark dwtDctSvd",
    "payload": "AGATE002",
    "metadata_keys": [
      "ai_generated",
      "generator",
      "model",
      "watermark"
    ],
    "default": "on",
    "detect": "agate.detect_watermark(img)"
  }
}