File size: 954 Bytes
d8735ee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{
  "_class_name": "ZImageTransformer2DModel",
  "_diffusers_version": "0.37.0.dev0",
  "_name_or_path": "/workspace/models/Tongyi-MAI/Z-Image/transformer",
  "all_f_patch_size": [
    1
  ],
  "all_patch_size": [
    2
  ],
  "axes_dims": [
    32,
    48,
    48
  ],
  "axes_lens": [
    1536,
    512,
    512
  ],
  "cap_feat_dim": 2560,
  "dim": 3840,
  "in_channels": 16,
  "n_heads": 30,
  "n_kv_heads": 30,
  "n_layers": 30,
  "n_refiner_layers": 2,
  "norm_eps": 1e-05,
  "qk_norm": true,
  "quantization_config": {
    "autoround_version": "0.13.0",
    "batch_size": 1,
    "bits": 4,
    "block_name_to_quantize": "noise_refiner,context_refiner,layers",
    "data_type": "int",
    "gradient_accumulate_steps": 8,
    "group_size": 128,
    "low_gpu_mem_usage": true,
    "packing_format": "auto_round:auto_gptq",
    "quant_method": "auto-round",
    "sym": true
  },
  "rope_theta": 256.0,
  "siglip_feat_dim": null,
  "t_scale": 1000.0
}