quiet-forest-34

FLUX.2 Klein 4B Step-500 EMA Diffusers Export

This repository contains a complete native Hugging Face Diffusers export of a trained FLUX.2 Klein Base 4B checkpoint. The public repository name is intentionally neutral; full experiment provenance is recorded below for reproducibility.

Provenance

• Source experiment: flux2_klein_base_4b_diffusionnft_dvreward_prompt_rubric_v4_3_1_geneval_16prompts_group14_7train_1dvreward_tp1_2node_512px_20step_cfg4_nrt
• Experiment variant: prompt-rubric v4.3.1, GenEval
• Formal run: flux2-v431-geneval-formal-505-142eef2e-a12
• Source run code revision: 142eef2e8ea30d30eaff17e3ddb2547f908fc3cc
• Source checkpoint: models/step_500.pt
• Global step: 500
• Exported weights: EMA
• Source checkpoint SHA-256: 2f69fbd3bddee06964755babb25b30b90dfc59c944eef01d6480f40966fa0e25
• Base model: black-forest-labs/FLUX.2-klein-base-4B
• Base revision: a3b4f4849157f664bdbc776fd7453c2783562f4d
• Training profile: 16 prompts, group size 14, 512 px, 20 rollout steps, CFG 4, 7 policy ranks and 1 DVReward replica per node across 2 nodes
• LoRA rank / alpha: 32 / 64

The EMA LoRA weights were merged into the native BF16 FLUX.2 transformer. The export contains 169 transformer tensors and 3,875,544,576 transformer parameters. DiffusionNFT old-policy training-state tensors were excluded from the inference artifact.

Format

• Pipeline: diffusers.Flux2KleinPipeline
• Transformer: diffusers.Flux2Transformer2DModel
• Precision: BF16
• Scheduler: 1,000 training timesteps, dynamic shifting enabled, deterministic sampling
• Transformer weights: 9 shards
• Text encoder weights: 9 shards
• Maximum shard size: below 1 GB

Verification

The export was strictly reloaded offline with stock Diffusers and generated a 512 x 512 smoke image using 4 inference steps, CFG 4, and seed 0. The merged transformer differs from Base in 60 tensors and 2,421,531,792 elements, with L2 delta 28.540123379650495 and maximum absolute delta 0.009033203125.

A same-seed Base render was also generated:

• Changed channel values: 762,311
• Mean absolute channel delta: 36.23766326904297
• Maximum absolute channel delta: 254

These checks verify that the exported transformer contains the trained Step-500 EMA weights rather than an unchanged Base transformer. verification.json, smoke images, the export manifest, and SHA-256 checksums are included.

Loading

import torch
from diffusers import Flux2KleinPipeline

pipe = Flux2KleinPipeline.from_pretrained(
    "kimi000/quiet-forest-34",
    torch_dtype=torch.bfloat16,
)
pipe.enable_model_cpu_offload()

image = pipe(
    prompt="A red cube beside a blue glass sphere.",
    height=1024,
    width=1024,
    guidance_scale=4.0,
    num_inference_steps=20,
).images[0]

See export_manifest.json and verification.json for machine-readable export details.