{
    "patch_size": 1,
    "in_channels": 128,
    "out_channels": null,
    "num_layers": 5,
    "num_single_layers": 20,
    "attention_head_dim": 128,
    "num_attention_heads": 24,
    "joint_attention_dim": 7680,
    "timestep_guidance_channels": 256,
    "mlp_ratio": 3.0,
    "axes_dims_rope": [
        32,
        32,
        32,
        32
    ],
    "rope_theta": 2000,
    "eps": 1e-06,
    "guidance_embeds": false,
    "_class_name": "Flux2Transformer2DModel",
    "_diffusers_version": "0.37.0.dev0",
    "_name_or_path": "/root/.cache/huggingface/hub/models--black-forest-labs--FLUX.2-klein-4B/snapshots/e7b7dc27f91deacad38e78976d1f2b499d76a294/transformer",
    "_FrozenDict__frozen": true
}