| { | |
| "model": "Z-Image-Turbo", | |
| "transformer": { | |
| "n_layers": 30, | |
| "n_refiner_layers": 2, | |
| "dim": 3840, | |
| "n_heads": 30, | |
| "n_kv_heads": 30, | |
| "in_channels": 16, | |
| "all_patch_size": [ | |
| 2 | |
| ], | |
| "axes_dims": [ | |
| 32, | |
| 48, | |
| 48 | |
| ], | |
| "axes_lens": [ | |
| 1536, | |
| 512, | |
| 512 | |
| ], | |
| "cap_feat_dim": 2560, | |
| "rope_theta": 256.0, | |
| "norm_eps": 1e-05, | |
| "qk_norm": true, | |
| "t_scale": 1000.0 | |
| }, | |
| "scheduler": { | |
| "kind": "FlowMatchEulerDiscreteScheduler", | |
| "shift": 3.0, | |
| "train_timesteps": 1000 | |
| }, | |
| "sampling": { | |
| "steps": 8, | |
| "guidance": 0.0 | |
| }, | |
| "quantization": { | |
| "format": "int8_tensorwise", | |
| "convrot": true, | |
| "convrot_groupsize": 256 | |
| } | |
| } | |