transformer/config.json
| 1 | { |
| 2 | "_class_name": "ZImageTransformer2DModel", |
| 3 | "_diffusers_version": "0.36.0.dev0", |
| 4 | "all_f_patch_size": [ |
| 5 | 1 |
| 6 | ], |
| 7 | "all_patch_size": [ |
| 8 | 2 |
| 9 | ], |
| 10 | "axes_dims": [ |
| 11 | 32, |
| 12 | 48, |
| 13 | 48 |
| 14 | ], |
| 15 | "axes_lens": [ |
| 16 | 1536, |
| 17 | 512, |
| 18 | 512 |
| 19 | ], |
| 20 | "cap_feat_dim": 2560, |
| 21 | "dim": 3840, |
| 22 | "in_channels": 16, |
| 23 | "n_heads": 30, |
| 24 | "n_kv_heads": 30, |
| 25 | "n_layers": 30, |
| 26 | "n_refiner_layers": 2, |
| 27 | "norm_eps": 1e-05, |
| 28 | "qk_norm": true, |
| 29 | "rope_theta": 256.0, |
| 30 | "t_scale": 1000.0 |
| 31 | } |
| 32 | |