| { | |
| "version": "flux-schnell", | |
| "params": { | |
| "in_channels": 64, | |
| "vec_in_dim": 768, | |
| "context_in_dim": 4096, | |
| "hidden_size": 3072, | |
| "mlp_ratio": 4.0, | |
| "num_heads": 24, | |
| "depth": 19, | |
| "depth_single_blocks": 38, | |
| "axes_dim": [16, 56, 56], | |
| "theta": 10000, | |
| "qkv_bias": true, | |
| "guidance_embed": false | |
| }, | |
| "ae_params": { | |
| "resolution": 256, | |
| "in_channels": 3, | |
| "ch": 128, | |
| "out_ch": 3, | |
| "ch_mult": [1, 2, 4, 4], | |
| "num_res_blocks": 2, | |
| "z_channels": 16, | |
| "scale_factor": 0.3611, | |
| "shift_factor": 0.1159 | |
| }, | |
| "ckpt_path": "flux1-schnell.safetensors", | |
| "ae_path": "ae.safetensors", | |
| "repo_id": "black-forest-labs/FLUX.1-dev", | |
| "repo_flow": "flux1-dev.sft", | |
| "repo_ae": "ae.sft", | |
| "text_enc_max_length": 512, | |
| "text_enc_path": "city96/t5-v1_1-xxl-encoder-bf16", | |
| "text_enc_device": "cuda:0", | |
| "ae_device": "cuda:0", | |
| "flux_device": "cuda:0", | |
| "flow_dtype": "float16", | |
| "ae_dtype": "bfloat16", | |
| "text_enc_dtype": "bfloat16", | |
| "flow_quantization_dtype": "qfloat8", | |
| "text_enc_quantization_dtype": "qfloat8", | |
| "compile_extras": true, | |
| "compile_blocks": true, | |
| "offload_text_encoder": false, | |
| "offload_vae": false, | |
| "offload_flow": false | |
| } | |