| { | |
| "_class_name": "Flux2Transformer2DModel", | |
| "_diffusers_version": "0.36.0.dev0", | |
| "attention_head_dim": 128, | |
| "axes_dims_rope": [ | |
| 32, | |
| 32, | |
| 32, | |
| 32 | |
| ], | |
| "eps": 1e-06, | |
| "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, | |
| "rope_theta": 2000, | |
| "timestep_guidance_channels": 256, | |
| "quantization_config": { | |
| "quant_method": "auto-round", | |
| "packing_format": "auto_round:llm_compressor", | |
| "bits": 8, | |
| "group_size": 32, | |
| "sym": true, | |
| "data_type": "mx_fp", | |
| "iters": 0, | |
| "model_free": true, | |
| "autoround_version": "0.17.0", | |
| "enable_quanted_input": false, | |
| "block_name_to_quantize": "transformer_blocks,single_transformer_blocks", | |
| "act_bits": 8, | |
| "act_data_type": "mx_fp", | |
| "act_dynamic": true, | |
| "act_group_size": 32, | |
| "act_sym": true, | |
| "extra_config": { | |
| "double_stream_modulation_img.linear": { | |
| "bits": 16, | |
| "data_type": "float", | |
| "act_bits": 16, | |
| "act_data_type": "float" | |
| }, | |
| "double_stream_modulation_txt.linear": { | |
| "bits": 16, | |
| "data_type": "float", | |
| "act_bits": 16, | |
| "act_data_type": "float" | |
| }, | |
| "single_stream_modulation.linear": { | |
| "bits": 16, | |
| "data_type": "float", | |
| "act_bits": 16, | |
| "act_data_type": "float" | |
| }, | |
| "norm_out.linear": { | |
| "bits": 16, | |
| "data_type": "float", | |
| "act_bits": 16, | |
| "act_data_type": "float" | |
| }, | |
| "proj_out": { | |
| "bits": 16, | |
| "data_type": "float", | |
| "act_bits": 16, | |
| "act_data_type": "float" | |
| } | |
| } | |
| } | |
| } |