{
  "architecture" : {
    "class" : "diffusion"
  },
  "capabilities" : [
    "imageOutput"
  ],
  "diffusion" : {
    "assetBytes" : {
      "denoiser0" : 1903883459,
      "denoiser1" : 1283968520,
      "denoiser2" : 1444859676,
      "textEncoder" : 3116676246,
      "vaeDecoder" : 101515325
    },
    "assetReshapes" : {
      "vaeDecoder" : false
    },
    "decoderAsset" : "vaeDecoder",
    "defaultSize" : {
      "height" : 1024,
      "width" : 1024
    },
    "denoiserAsset" : "denoiser",
    "denoiserSegments" : [
      "denoiser0",
      "denoiser1",
      "denoiser2"
    ],
    "denoiserStreaming" : {
      "minImageTokens" : 2305
    },
    "expectFrequentReshapes" : true,
    "guidance" : false,
    "latentChannels" : 128,
    "maxSteps" : 50,
    "pixelsPerToken" : 16,
    "prompt" : {
      "maxLength" : 512,
      "padToken" : 151643,
      "padding" : "right",
      "template" : "chat",
      "thinking" : "off"
    },
    "rotary" : {
      "axes" : [
        32,
        32,
        32,
        32
      ],
      "theta" : 2000
    },
    "sigmaShift" : {
      "highIntercept" : 0.45666666,
      "highSlope" : 0.00016927,
      "highSteps" : 200,
      "lowIntercept" : 1.89833333,
      "lowSlope" : 8.73809524e-05,
      "lowSteps" : 10,
      "rule" : "empirical",
      "tokenThreshold" : 4300
    },
    "sizes" : [
      {
        "height" : 1024,
        "width" : 1024
      },
      {
        "height" : 512,
        "width" : 512
      },
      {
        "height" : 768,
        "width" : 768
      },
      {
        "height" : 768,
        "width" : 1024
      },
      {
        "height" : 1024,
        "width" : 768
      },
      {
        "height" : 768,
        "width" : 1344
      },
      {
        "height" : 1344,
        "width" : 768
      }
    ],
    "steps" : 4,
    "textEncoderAsset" : "textEncoder",
    "trainTimesteps" : 1000
  },
  "driver" : "diffusion",
  "functions" : {
    "denoiser0/segment" : {
      "role" : "denoiseSegment"
    },
    "denoiser1/segment" : {
      "role" : "denoiseSegment"
    },
    "denoiser2/segment" : {
      "role" : "denoiseSegment"
    },
    "textEncoder/encode" : {
      "role" : "encodeText"
    },
    "vaeDecoder/decode" : {
      "role" : "decodeLatents"
    }
  },
  "inputs" : {
    "attention_mask" : {
      "bind" : "tokenMask",
      "dtype" : "int32"
    },
    "cond" : {
      "bind" : "carry",
      "dtype" : "float32"
    },
    "image_1" : {
      "bind" : "carry",
      "dtype" : "float32"
    },
    "image_2" : {
      "bind" : "carry",
      "dtype" : "float32"
    },
    "input_ids" : {
      "bind" : "tokens",
      "dtype" : "int32"
    },
    "latents" : {
      "bind" : "latents",
      "dtype" : "float16"
    },
    "position_ids" : {
      "bind" : "positions",
      "dtype" : "int32",
      "mode" : "running"
    },
    "rope_image" : {
      "bind" : "imageRotary",
      "dtype" : "float32"
    },
    "rope_text" : {
      "bind" : "textRotary",
      "dtype" : "float32"
    },
    "sigma" : {
      "bind" : "timestep",
      "dtype" : "float32"
    },
    "text_1" : {
      "bind" : "carry",
      "dtype" : "float32"
    },
    "text_2" : {
      "bind" : "carry",
      "dtype" : "float32"
    },
    "text_embeds" : {
      "bind" : "textEmbeddings",
      "dtype" : "float16"
    }
  },
  "memory" : {
    "requiresIncreasedMemoryLimit" : false,
    "weightsBytes" : 7850903226
  },
  "outputs" : {
    "cond" : {
      "dtype" : "float32",
      "role" : "carry"
    },
    "image" : {
      "dtype" : "float32",
      "role" : "pixels"
    },
    "image_1" : {
      "dtype" : "float32",
      "role" : "carry"
    },
    "image_2" : {
      "dtype" : "float32",
      "role" : "carry"
    },
    "text_1" : {
      "dtype" : "float32",
      "role" : "carry"
    },
    "text_2" : {
      "dtype" : "float32",
      "role" : "carry"
    },
    "text_embeds" : {
      "dtype" : "float16",
      "role" : "textEmbeddings"
    },
    "velocity" : {
      "dtype" : "float16",
      "role" : "velocity"
    }
  },
  "provenance" : {
    "exportedOn" : {
      "date" : "2026-09-24",
      "os" : "26A428"
    },
    "files" : {
      "flux2_klein_4b-denoiser0.aimodel/main.hash" : "sha256:97757c21208abde90b7d998157fdc168584e6c38decb55db4196ade2effafe4c",
      "flux2_klein_4b-denoiser0.aimodel/main.mlirb" : "sha256:3107b5b9dece7f1390476d5192e0fe8548295712eaa44cee2d20dc2d044c9f02",
      "flux2_klein_4b-denoiser0.aimodel/metadata.json" : "sha256:f918e7cdf3fa8073d814d254566c73a964fd8c809b61edeb5840c0afca496747",
      "flux2_klein_4b-denoiser1.aimodel/main.hash" : "sha256:248e47f9d34f961afbfac3add32c4fb09f21235277f53a151f3ac12b593aa8e5",
      "flux2_klein_4b-denoiser1.aimodel/main.mlirb" : "sha256:3e28f1a124c3516c717413887868e07407131888a36e0da98d1d70d0305286d0",
      "flux2_klein_4b-denoiser1.aimodel/metadata.json" : "sha256:cd4e6d41f9ada58120e9098d6983722bfd5e713ede9ed8bb80c7904f35c83f7a",
      "flux2_klein_4b-denoiser2.aimodel/main.hash" : "sha256:dd1ff023ac1dab94b4a77b35b98f77aa6cc99ca3f26e59c45598eaf2969e9673",
      "flux2_klein_4b-denoiser2.aimodel/main.mlirb" : "sha256:04db4ef155fc2245b608827573d91b542aa7da7294ab25bb235ecca1d2d3fa96",
      "flux2_klein_4b-denoiser2.aimodel/metadata.json" : "sha256:783dad44a18759c753c45df4aa697560a3cd96300f40ae618a9626b58ff769ba",
      "flux2_klein_4b-text.aimodel/main.hash" : "sha256:a0f5f9f3bacc222702e91fe751f7f24c21075d0e67ef0bafac16f777c4285851",
      "flux2_klein_4b-text.aimodel/main.mlirb" : "sha256:f6a2c232ce0e9bc13ea9cfa8d723f2afb4910ffa39df676fc84f2a9b2370898d",
      "flux2_klein_4b-text.aimodel/metadata.json" : "sha256:4103be5cab3f8d4b0b8fb37fbe711f20e25b6502581df0bda67ff49163d9df97",
      "flux2_klein_4b-vae.aimodel/main.hash" : "sha256:ee933b0e2b94b8326568d8a36792d8d4c135f85bafb6192d543ed2424c377b47",
      "flux2_klein_4b-vae.aimodel/main.mlirb" : "sha256:445d2a94d7590d3c09f24bf736bac5ccfdd5153476ae97a863665afbc6a8ea2d",
      "flux2_klein_4b-vae.aimodel/metadata.json" : "sha256:bb7093e0eaaf98bc4f7114c3946c124258e0a2d62970d870b63fe4ea1b0b010a",
      "metadata.json" : "sha256:29ab9971d5c35c18ce4bbe675981b278a32435447223be1a325269e137a9bf09",
      "tokenizer/added_tokens.json" : "sha256:c0284b582e14987fbd3d5a2cb2bd139084371ed9acbae488829a1c900833c680",
      "tokenizer/chat_template.jinja" : "sha256:a55ee1b1660128b7098723e0abcd92caa0788061051c62d51cbe87d9cf1974d8",
      "tokenizer/merges.txt" : "sha256:8831e4f1a044471340f7c0a83d7bd71306a5b867e95fd870f74d0c5308a904d5",
      "tokenizer/special_tokens_map.json" : "sha256:76862e765266b85aa9459767e33cbaf13970f327a0e88d1c65846c2ddd3a1ecd",
      "tokenizer/tokenizer.json" : "sha256:aeb13307a71acd8fe81861d94ad54ab689df773318809eed3cbe794b4492dae4",
      "tokenizer/tokenizer_config.json" : "sha256:443bfa629eb16387a12edbf92a76f6a6f10b2af3b53d87ba1550adfcf45f7fa0",
      "tokenizer/vocab.json" : "sha256:ca10d7e9fb3ed18575dd1e277a2579c16d108e32f27439684afa0e10b1440910"
    },
    "forge" : "0.1.0",
    "recipe" : "flux2-klein-4b@3",
    "source" : {
      "hfRepo" : "black-forest-labs/FLUX.2-klein-4B",
      "revision" : "e7b7dc27f91deacad38e78976d1f2b499d76a294"
    }
  },
  "quantization" : {
    "exceptions" : {
      "singles.*.proj_out" : "none"
    },
    "notes" : "weight-only PTQ on the torch module (coreai-opt eager Quantizer); exceptions resolve first-match in declaration order, then default; modules: int8-linear-perchannel ×205, none ×20",
    "scheme" : "int8-linear-perchannel",
    "weightBits" : 8
  },
  "schema" : 1,
  "target" : {
    "arch" : "any",
    "backend" : "coreai",
    "compiled" : false,
    "compute" : "gpu",
    "minOS" : "27.0",
    "platform" : "macos"
  },
  "text" : {
    "chatTemplate" : "embedded",
    "thinking" : {
      "close" : "</think>",
      "open" : "<think>",
      "style" : "tags",
      "toggle" : "templateKwarg:enable_thinking"
    },
    "tokenizer" : "tokenizer/"
  }
}