| { |
| "_class_name": "Flux2Transformer2DModel", |
| "_diffusers_version": "0.37.1", |
| "_name_or_path": "/root/.cache/huggingface/hub/models--black-forest-labs--FLUX.2-klein-9B/snapshots/92196c8e11f7b6cf2b7493e037d8c5345c559216/transformer", |
| "attention_head_dim": 128, |
| "axes_dims_rope": [ |
| 32, |
| 32, |
| 32, |
| 32 |
| ], |
| "eps": 1e-06, |
| "guidance_embeds": false, |
| "in_channels": 128, |
| "joint_attention_dim": 12288, |
| "mlp_ratio": 3.0, |
| "num_attention_heads": 32, |
| "num_layers": 8, |
| "num_single_layers": 24, |
| "out_channels": null, |
| "patch_size": 1, |
| "quantization_config": { |
| "add_skip_keys": true, |
| "dequantize_fp32": true, |
| "dynamic_loss_threshold": 0.1, |
| "group_size": 0, |
| "is_integer": true, |
| "is_training": false, |
| "modules_dtype_dict": { |
| "int3": [ |
| "transformer_blocks.0.attn.to_v.weight", |
| "transformer_blocks.0.attn.to_out.0.weight", |
| "transformer_blocks.0.attn.add_q_proj.weight", |
| "transformer_blocks.0.attn.add_k_proj.weight", |
| "transformer_blocks.0.attn.add_v_proj.weight", |
| "transformer_blocks.0.attn.to_add_out.weight", |
| "transformer_blocks.0.ff.linear_in.weight", |
| "transformer_blocks.0.ff.linear_out.weight", |
| "transformer_blocks.0.ff_context.linear_in.weight", |
| "transformer_blocks.0.ff_context.linear_out.weight", |
| "transformer_blocks.1.attn.to_q.weight", |
| "transformer_blocks.1.attn.to_k.weight", |
| "transformer_blocks.1.attn.to_v.weight", |
| "transformer_blocks.1.attn.to_out.0.weight", |
| "transformer_blocks.1.attn.add_q_proj.weight", |
| "transformer_blocks.1.attn.add_k_proj.weight", |
| "transformer_blocks.1.attn.add_v_proj.weight", |
| "transformer_blocks.1.attn.to_add_out.weight", |
| "transformer_blocks.1.ff.linear_in.weight", |
| "transformer_blocks.1.ff.linear_out.weight", |
| "transformer_blocks.1.ff_context.linear_in.weight", |
| "transformer_blocks.1.ff_context.linear_out.weight", |
| "transformer_blocks.2.attn.to_q.weight", |
| "transformer_blocks.2.attn.to_k.weight", |
| "transformer_blocks.2.attn.to_v.weight", |
| "transformer_blocks.2.attn.to_out.0.weight", |
| "transformer_blocks.2.attn.add_q_proj.weight", |
| "transformer_blocks.2.attn.add_k_proj.weight", |
| "transformer_blocks.2.attn.add_v_proj.weight", |
| "transformer_blocks.2.attn.to_add_out.weight", |
| "transformer_blocks.2.ff.linear_in.weight", |
| "transformer_blocks.2.ff.linear_out.weight", |
| "transformer_blocks.2.ff_context.linear_in.weight", |
| "transformer_blocks.2.ff_context.linear_out.weight", |
| "transformer_blocks.3.attn.to_q.weight", |
| "transformer_blocks.3.attn.to_k.weight", |
| "transformer_blocks.3.attn.to_v.weight", |
| "transformer_blocks.3.attn.to_out.0.weight", |
| "transformer_blocks.3.attn.add_q_proj.weight", |
| "transformer_blocks.3.attn.add_k_proj.weight", |
| "transformer_blocks.3.attn.add_v_proj.weight", |
| "transformer_blocks.3.attn.to_add_out.weight", |
| "transformer_blocks.3.ff.linear_in.weight", |
| "transformer_blocks.3.ff.linear_out.weight", |
| "transformer_blocks.3.ff_context.linear_in.weight", |
| "transformer_blocks.3.ff_context.linear_out.weight", |
| "transformer_blocks.4.attn.to_q.weight", |
| "transformer_blocks.4.attn.to_k.weight", |
| "transformer_blocks.4.attn.to_v.weight", |
| "transformer_blocks.4.attn.to_out.0.weight", |
| "transformer_blocks.4.attn.add_q_proj.weight", |
| "transformer_blocks.4.attn.add_k_proj.weight", |
| "transformer_blocks.4.attn.add_v_proj.weight", |
| "transformer_blocks.4.attn.to_add_out.weight", |
| "transformer_blocks.4.ff.linear_in.weight", |
| "transformer_blocks.4.ff.linear_out.weight", |
| "transformer_blocks.4.ff_context.linear_in.weight", |
| "transformer_blocks.4.ff_context.linear_out.weight", |
| "transformer_blocks.5.attn.to_q.weight", |
| "transformer_blocks.5.attn.to_k.weight", |
| "transformer_blocks.5.attn.to_v.weight", |
| "transformer_blocks.5.attn.to_out.0.weight", |
| "transformer_blocks.5.attn.add_q_proj.weight", |
| "transformer_blocks.5.attn.add_k_proj.weight", |
| "transformer_blocks.5.attn.add_v_proj.weight", |
| "transformer_blocks.5.attn.to_add_out.weight", |
| "transformer_blocks.5.ff.linear_in.weight", |
| "transformer_blocks.5.ff.linear_out.weight", |
| "transformer_blocks.5.ff_context.linear_in.weight", |
| "transformer_blocks.5.ff_context.linear_out.weight", |
| "transformer_blocks.6.attn.to_q.weight", |
| "transformer_blocks.6.attn.to_k.weight", |
| "transformer_blocks.6.attn.to_v.weight", |
| "transformer_blocks.6.attn.to_out.0.weight", |
| "transformer_blocks.6.attn.add_q_proj.weight", |
| "transformer_blocks.6.attn.add_k_proj.weight", |
| "transformer_blocks.6.attn.add_v_proj.weight", |
| "transformer_blocks.6.attn.to_add_out.weight", |
| "transformer_blocks.6.ff.linear_in.weight", |
| "transformer_blocks.6.ff.linear_out.weight", |
| "transformer_blocks.6.ff_context.linear_in.weight", |
| "transformer_blocks.6.ff_context.linear_out.weight", |
| "transformer_blocks.7.attn.to_q.weight", |
| "transformer_blocks.7.attn.to_k.weight", |
| "transformer_blocks.7.attn.to_v.weight", |
| "transformer_blocks.7.attn.to_out.0.weight", |
| "transformer_blocks.7.attn.add_q_proj.weight", |
| "transformer_blocks.7.attn.add_k_proj.weight", |
| "transformer_blocks.7.attn.add_v_proj.weight", |
| "transformer_blocks.7.attn.to_add_out.weight", |
| "transformer_blocks.7.ff.linear_in.weight", |
| "transformer_blocks.7.ff.linear_out.weight", |
| "transformer_blocks.7.ff_context.linear_in.weight", |
| "transformer_blocks.7.ff_context.linear_out.weight", |
| "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.0.attn.to_out.weight", |
| "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.1.attn.to_out.weight", |
| "single_transformer_blocks.2.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.2.attn.to_out.weight", |
| "single_transformer_blocks.3.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.3.attn.to_out.weight", |
| "single_transformer_blocks.4.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.4.attn.to_out.weight", |
| "single_transformer_blocks.5.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.5.attn.to_out.weight", |
| "single_transformer_blocks.6.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.6.attn.to_out.weight", |
| "single_transformer_blocks.7.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.7.attn.to_out.weight", |
| "single_transformer_blocks.8.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.8.attn.to_out.weight", |
| "single_transformer_blocks.9.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.9.attn.to_out.weight", |
| "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.10.attn.to_out.weight", |
| "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.11.attn.to_out.weight", |
| "single_transformer_blocks.12.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.12.attn.to_out.weight", |
| "single_transformer_blocks.13.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.13.attn.to_out.weight", |
| "single_transformer_blocks.14.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.14.attn.to_out.weight", |
| "single_transformer_blocks.15.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.15.attn.to_out.weight", |
| "single_transformer_blocks.16.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.16.attn.to_out.weight", |
| "single_transformer_blocks.17.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.17.attn.to_out.weight", |
| "single_transformer_blocks.18.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.18.attn.to_out.weight", |
| "single_transformer_blocks.19.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.19.attn.to_out.weight", |
| "single_transformer_blocks.20.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.20.attn.to_out.weight", |
| "single_transformer_blocks.21.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.21.attn.to_out.weight", |
| "single_transformer_blocks.22.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.22.attn.to_out.weight", |
| "single_transformer_blocks.23.attn.to_qkv_mlp_proj.weight", |
| "single_transformer_blocks.23.attn.to_out.weight" |
| ], |
| "uint2": [ |
| "transformer_blocks.0.attn.to_q.weight", |
| "transformer_blocks.0.attn.to_k.weight" |
| ] |
| }, |
| "modules_quant_config": {}, |
| "modules_to_not_convert": [ |
| "double_stream_modulation_txt", |
| "time_guidance_embed", |
| "x_embedder", |
| ".proj_out", |
| "single_stream_modulation", |
| "context_embedder", |
| "norm_out", |
| "double_stream_modulation_img" |
| ], |
| "non_blocking": false, |
| "quant_conv": false, |
| "quant_embedding": false, |
| "quant_method": "sdnq", |
| "quantization_device": "cuda", |
| "quantized_matmul_dtype": null, |
| "return_device": "cpu", |
| "sdnq_version": "0.1.8", |
| "svd_rank": 32, |
| "svd_steps": 16, |
| "use_dynamic_quantization": true, |
| "use_grad_ckpt": true, |
| "use_quantized_matmul": false, |
| "use_quantized_matmul_conv": false, |
| "use_static_quantization": true, |
| "use_stochastic_rounding": false, |
| "use_svd": true, |
| "weights_dtype": "uint2" |
| }, |
| "rope_theta": 2000, |
| "timestep_guidance_channels": 256 |
| } |
|
|
