| { |
| "_class_name": "Flux2Transformer2DModel", |
| "_diffusers_version": "0.36.0", |
| "_name_or_path": "black-forest-labs/FLUX.2-dev", |
| "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, |
| "quantization_config": { |
| "include_input_output_embeddings": false, |
| "modules_to_not_convert": null, |
| "quant_method": "torchao", |
| "quant_type": { |
| "default": { |
| "_data": { |
| "activation_dtype": { |
| "_data": "float8_e4m3fn", |
| "_type": "torch.dtype" |
| }, |
| "activation_value_lb": null, |
| "activation_value_ub": null, |
| "granularity": [ |
| { |
| "_data": {}, |
| "_type": "PerTensor", |
| "_version": 1 |
| }, |
| { |
| "_data": {}, |
| "_type": "PerTensor", |
| "_version": 1 |
| } |
| ], |
| "kernel_preference": { |
| "_data": "AUTO", |
| "_type": "KernelPreference" |
| }, |
| "mm_config": { |
| "_data": { |
| "emulate": false, |
| "pad_inner_dim": false, |
| "use_fast_accum": true |
| }, |
| "_type": "Float8MMConfig", |
| "_version": 1 |
| }, |
| "set_inductor_config": true, |
| "weight_dtype": { |
| "_data": "float8_e4m3fn", |
| "_type": "torch.dtype" |
| } |
| }, |
| "_type": "Float8DynamicActivationFloat8WeightConfig", |
| "_version": 2 |
| } |
| }, |
| "quant_type_kwargs": {}, |
| "untie_embedding_weights": false |
| }, |
| "rope_theta": 2000, |
| "timestep_guidance_channels": 256 |
| } |
|
|
