{ "_class_name": "Flux2Transformer2DModel", "_diffusers_version": "0.38.0.dev0", "_name_or_path": "black-forest-labs/FLUX.2-dev", "attention_head_dim": 128, "axes_dims_rope": [ 32, 32, 32, 32 ], "eps": 1e-06, "guidance_embeds": true, "in_channels": 128, "joint_attention_dim": 15360, "mlp_ratio": 3.0, "num_attention_heads": 48, "num_layers": 8, "num_single_layers": 48, "out_channels": null, "patch_size": 1, "quantization_config": { "add_skip_keys": true, "dequantize_fp32": false, "dynamic_loss_threshold": 0.0001, "group_size": 0, "hadamard_group_size": 128, "is_integer": true, "is_training": false, "modules_dtype_dict": { "int8": [ "transformer_blocks.0.attn.add_k_proj.weight", "transformer_blocks.0.attn.add_q_proj.weight", "transformer_blocks.0.attn.add_v_proj.weight", "transformer_blocks.0.attn.to_add_out.weight", "transformer_blocks.0.attn.to_k.weight", "transformer_blocks.0.attn.to_out.0.weight", "transformer_blocks.0.attn.to_q.weight", "transformer_blocks.0.attn.to_v.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.add_k_proj.weight", "transformer_blocks.1.attn.add_q_proj.weight", "transformer_blocks.1.attn.add_v_proj.weight", "transformer_blocks.1.attn.to_add_out.weight", "transformer_blocks.1.attn.to_k.weight", "transformer_blocks.1.attn.to_out.0.weight", "transformer_blocks.1.attn.to_q.weight", "transformer_blocks.1.attn.to_v.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.add_k_proj.weight", "transformer_blocks.2.attn.add_q_proj.weight", "transformer_blocks.2.attn.add_v_proj.weight", "transformer_blocks.2.attn.to_add_out.weight", "transformer_blocks.2.attn.to_k.weight", "transformer_blocks.2.attn.to_out.0.weight", "transformer_blocks.2.attn.to_q.weight", "transformer_blocks.2.attn.to_v.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.add_k_proj.weight", "transformer_blocks.3.attn.add_q_proj.weight", "transformer_blocks.3.attn.add_v_proj.weight", "transformer_blocks.3.attn.to_add_out.weight", "transformer_blocks.3.attn.to_k.weight", "transformer_blocks.3.attn.to_out.0.weight", "transformer_blocks.3.attn.to_q.weight", "transformer_blocks.3.attn.to_v.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.add_k_proj.weight", "transformer_blocks.4.attn.add_q_proj.weight", "transformer_blocks.4.attn.add_v_proj.weight", "transformer_blocks.4.attn.to_add_out.weight", "transformer_blocks.4.attn.to_k.weight", "transformer_blocks.4.attn.to_out.0.weight", "transformer_blocks.4.attn.to_q.weight", "transformer_blocks.4.attn.to_v.weight", "single_transformer_blocks.0.attn.to_out.weight", "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.1.attn.to_out.weight", "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.2.attn.to_qkv_mlp_proj.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.add_k_proj.weight", "transformer_blocks.5.attn.add_q_proj.weight", "transformer_blocks.5.attn.add_v_proj.weight", "transformer_blocks.5.attn.to_add_out.weight", "transformer_blocks.5.attn.to_k.weight", "transformer_blocks.5.attn.to_out.0.weight", "transformer_blocks.5.attn.to_q.weight", "transformer_blocks.5.attn.to_v.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.add_k_proj.weight", "transformer_blocks.6.attn.add_q_proj.weight", "transformer_blocks.6.attn.add_v_proj.weight", "transformer_blocks.6.attn.to_add_out.weight", "transformer_blocks.6.attn.to_k.weight", "transformer_blocks.6.attn.to_out.0.weight", "transformer_blocks.6.attn.to_q.weight", "transformer_blocks.6.attn.to_v.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.add_k_proj.weight", "transformer_blocks.7.attn.add_q_proj.weight", "transformer_blocks.7.attn.add_v_proj.weight", "transformer_blocks.7.attn.to_add_out.weight", "transformer_blocks.7.attn.to_k.weight", "transformer_blocks.7.attn.to_out.0.weight", "transformer_blocks.7.attn.to_q.weight", "transformer_blocks.7.attn.to_v.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.10.attn.to_out.weight", "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.11.attn.to_out.weight", "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.12.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.2.attn.to_out.weight", "single_transformer_blocks.3.attn.to_out.weight", "single_transformer_blocks.3.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.4.attn.to_out.weight", "single_transformer_blocks.4.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.5.attn.to_out.weight", "single_transformer_blocks.5.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.6.attn.to_out.weight", "single_transformer_blocks.6.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.7.attn.to_out.weight", "single_transformer_blocks.7.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.8.attn.to_out.weight", "single_transformer_blocks.8.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.9.attn.to_out.weight", "single_transformer_blocks.9.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.12.attn.to_out.weight", "single_transformer_blocks.13.attn.to_out.weight", "single_transformer_blocks.13.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.14.attn.to_out.weight", "single_transformer_blocks.14.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.15.attn.to_out.weight", "single_transformer_blocks.15.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.16.attn.to_out.weight", "single_transformer_blocks.16.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.17.attn.to_out.weight", "single_transformer_blocks.17.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.18.attn.to_out.weight", "single_transformer_blocks.18.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.19.attn.to_out.weight", "single_transformer_blocks.19.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.20.attn.to_out.weight", "single_transformer_blocks.20.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.21.attn.to_out.weight", "single_transformer_blocks.21.attn.to_qkv_mlp_proj.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_out.weight", "single_transformer_blocks.23.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.24.attn.to_out.weight", "single_transformer_blocks.24.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.25.attn.to_out.weight", "single_transformer_blocks.25.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.26.attn.to_out.weight", "single_transformer_blocks.26.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.27.attn.to_out.weight", "single_transformer_blocks.27.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.28.attn.to_out.weight", "single_transformer_blocks.28.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.29.attn.to_out.weight", "single_transformer_blocks.29.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.30.attn.to_out.weight", "single_transformer_blocks.30.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.31.attn.to_out.weight", "single_transformer_blocks.31.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.32.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.32.attn.to_out.weight", "single_transformer_blocks.33.attn.to_out.weight", "single_transformer_blocks.33.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.34.attn.to_out.weight", "single_transformer_blocks.34.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.35.attn.to_out.weight", "single_transformer_blocks.35.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.36.attn.to_out.weight", "single_transformer_blocks.36.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.37.attn.to_out.weight", "single_transformer_blocks.37.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.38.attn.to_out.weight", "single_transformer_blocks.38.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.39.attn.to_out.weight", "single_transformer_blocks.39.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.40.attn.to_out.weight", "single_transformer_blocks.40.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.41.attn.to_out.weight", "single_transformer_blocks.41.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.42.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.42.attn.to_out.weight", "single_transformer_blocks.43.attn.to_out.weight", "single_transformer_blocks.43.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.44.attn.to_out.weight", "single_transformer_blocks.44.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.45.attn.to_out.weight", "single_transformer_blocks.45.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.46.attn.to_out.weight", "single_transformer_blocks.46.attn.to_qkv_mlp_proj.weight", "single_transformer_blocks.47.attn.to_out.weight", "single_transformer_blocks.47.attn.to_qkv_mlp_proj.weight" ] }, "modules_quant_config": {}, "modules_to_not_convert": [ "double_stream_modulation_img", "double_stream_modulation_txt", "x_embedder", "time_guidance_embed", "norm_out", "single_stream_modulation", ".proj_out", "context_embedder", "transformer_blocks.0.attn.norm_added_k.weight", "transformer_blocks.0.attn.norm_added_q.weight", "transformer_blocks.0.attn.norm_k.weight", "transformer_blocks.0.attn.norm_q.weight", "transformer_blocks.1.attn.norm_added_k.weight", "transformer_blocks.1.attn.norm_added_q.weight", "transformer_blocks.1.attn.norm_k.weight", "transformer_blocks.1.attn.norm_q.weight", "transformer_blocks.2.attn.norm_added_k.weight", "transformer_blocks.2.attn.norm_added_q.weight", "transformer_blocks.2.attn.norm_k.weight", "transformer_blocks.2.attn.norm_q.weight", "transformer_blocks.3.attn.norm_added_k.weight", "transformer_blocks.3.attn.norm_added_q.weight", "transformer_blocks.3.attn.norm_k.weight", "transformer_blocks.3.attn.norm_q.weight", "transformer_blocks.4.attn.norm_added_k.weight", "transformer_blocks.4.attn.norm_added_q.weight", "transformer_blocks.4.attn.norm_k.weight", "transformer_blocks.4.attn.norm_q.weight", "single_transformer_blocks.0.attn.norm_k.weight", "single_transformer_blocks.0.attn.norm_q.weight", "single_transformer_blocks.1.attn.norm_k.weight", "single_transformer_blocks.1.attn.norm_q.weight", "single_transformer_blocks.2.attn.norm_k.weight", "single_transformer_blocks.2.attn.norm_q.weight", "transformer_blocks.5.attn.norm_added_k.weight", "transformer_blocks.5.attn.norm_added_q.weight", "transformer_blocks.5.attn.norm_k.weight", "transformer_blocks.5.attn.norm_q.weight", "transformer_blocks.6.attn.norm_added_k.weight", "transformer_blocks.6.attn.norm_added_q.weight", "transformer_blocks.6.attn.norm_k.weight", "transformer_blocks.6.attn.norm_q.weight", "transformer_blocks.7.attn.norm_added_k.weight", "transformer_blocks.7.attn.norm_added_q.weight", "transformer_blocks.7.attn.norm_k.weight", "transformer_blocks.7.attn.norm_q.weight", "single_transformer_blocks.10.attn.norm_k.weight", "single_transformer_blocks.10.attn.norm_q.weight", "single_transformer_blocks.11.attn.norm_k.weight", "single_transformer_blocks.11.attn.norm_q.weight", "single_transformer_blocks.12.attn.norm_k.weight", "single_transformer_blocks.12.attn.norm_q.weight", "single_transformer_blocks.3.attn.norm_k.weight", "single_transformer_blocks.3.attn.norm_q.weight", "single_transformer_blocks.4.attn.norm_k.weight", "single_transformer_blocks.4.attn.norm_q.weight", "single_transformer_blocks.5.attn.norm_k.weight", "single_transformer_blocks.5.attn.norm_q.weight", "single_transformer_blocks.6.attn.norm_k.weight", "single_transformer_blocks.6.attn.norm_q.weight", "single_transformer_blocks.7.attn.norm_k.weight", "single_transformer_blocks.7.attn.norm_q.weight", "single_transformer_blocks.8.attn.norm_k.weight", "single_transformer_blocks.8.attn.norm_q.weight", "single_transformer_blocks.9.attn.norm_k.weight", "single_transformer_blocks.9.attn.norm_q.weight", "single_transformer_blocks.13.attn.norm_k.weight", "single_transformer_blocks.13.attn.norm_q.weight", "single_transformer_blocks.14.attn.norm_k.weight", "single_transformer_blocks.14.attn.norm_q.weight", "single_transformer_blocks.15.attn.norm_k.weight", "single_transformer_blocks.15.attn.norm_q.weight", "single_transformer_blocks.16.attn.norm_k.weight", "single_transformer_blocks.16.attn.norm_q.weight", "single_transformer_blocks.17.attn.norm_k.weight", "single_transformer_blocks.17.attn.norm_q.weight", "single_transformer_blocks.18.attn.norm_k.weight", "single_transformer_blocks.18.attn.norm_q.weight", "single_transformer_blocks.19.attn.norm_k.weight", "single_transformer_blocks.19.attn.norm_q.weight", "single_transformer_blocks.20.attn.norm_k.weight", "single_transformer_blocks.20.attn.norm_q.weight", "single_transformer_blocks.21.attn.norm_k.weight", "single_transformer_blocks.21.attn.norm_q.weight", "single_transformer_blocks.22.attn.norm_k.weight", "single_transformer_blocks.22.attn.norm_q.weight", "single_transformer_blocks.23.attn.norm_k.weight", "single_transformer_blocks.23.attn.norm_q.weight", "single_transformer_blocks.24.attn.norm_k.weight", "single_transformer_blocks.24.attn.norm_q.weight", "single_transformer_blocks.25.attn.norm_k.weight", "single_transformer_blocks.25.attn.norm_q.weight", "single_transformer_blocks.26.attn.norm_k.weight", "single_transformer_blocks.26.attn.norm_q.weight", "single_transformer_blocks.27.attn.norm_k.weight", "single_transformer_blocks.27.attn.norm_q.weight", "single_transformer_blocks.28.attn.norm_k.weight", "single_transformer_blocks.28.attn.norm_q.weight", "single_transformer_blocks.29.attn.norm_k.weight", "single_transformer_blocks.29.attn.norm_q.weight", "single_transformer_blocks.30.attn.norm_k.weight", "single_transformer_blocks.30.attn.norm_q.weight", "single_transformer_blocks.31.attn.norm_k.weight", "single_transformer_blocks.31.attn.norm_q.weight", "single_transformer_blocks.32.attn.norm_k.weight", "single_transformer_blocks.32.attn.norm_q.weight", "single_transformer_blocks.33.attn.norm_k.weight", "single_transformer_blocks.33.attn.norm_q.weight", "single_transformer_blocks.34.attn.norm_k.weight", "single_transformer_blocks.34.attn.norm_q.weight", "single_transformer_blocks.35.attn.norm_k.weight", "single_transformer_blocks.35.attn.norm_q.weight", "single_transformer_blocks.36.attn.norm_k.weight", "single_transformer_blocks.36.attn.norm_q.weight", "single_transformer_blocks.37.attn.norm_k.weight", "single_transformer_blocks.37.attn.norm_q.weight", "single_transformer_blocks.38.attn.norm_k.weight", "single_transformer_blocks.38.attn.norm_q.weight", "single_transformer_blocks.39.attn.norm_k.weight", "single_transformer_blocks.39.attn.norm_q.weight", "single_transformer_blocks.40.attn.norm_k.weight", "single_transformer_blocks.40.attn.norm_q.weight", "single_transformer_blocks.41.attn.norm_k.weight", "single_transformer_blocks.41.attn.norm_q.weight", "single_transformer_blocks.42.attn.norm_k.weight", "single_transformer_blocks.42.attn.norm_q.weight", "single_transformer_blocks.43.attn.norm_k.weight", "single_transformer_blocks.43.attn.norm_q.weight", "single_transformer_blocks.44.attn.norm_k.weight", "single_transformer_blocks.44.attn.norm_q.weight", "single_transformer_blocks.45.attn.norm_k.weight", "single_transformer_blocks.45.attn.norm_q.weight", "single_transformer_blocks.46.attn.norm_k.weight", "single_transformer_blocks.46.attn.norm_q.weight", "single_transformer_blocks.47.attn.norm_k.weight", "single_transformer_blocks.47.attn.norm_q.weight" ], "modules_to_not_use_matmul": [], "non_blocking": false, "quant_conv": false, "quant_embedding": false, "quant_method": "sdnq", "quantization_device": null, "quantized_matmul_dtype": null, "return_device": null, "sdnq_version": "0.1.9", "svd_rank": 32, "svd_steps": 8, "use_dynamic_quantization": true, "use_grad_ckpt": true, "use_hadamard": true, "use_quantized_matmul": false, "use_quantized_matmul_conv": false, "use_static_quantization": true, "use_stochastic_rounding": false, "use_svd": false, "weights_dtype": "int8" }, "rope_theta": 2000, "timestep_guidance_channels": 256 }