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