adhikjoshi commited on
Commit
117b828
·
verified ·
1 Parent(s): ea1b669

W4A4 int4 checkpoint, SVDQuant + GPTQ, LPIPS 0.1836

Browse files
.gitattributes CHANGED
@@ -33,3 +33,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ samples/flat.png filter=lfs diff=lfs merge=lfs -text
37
+ samples/landscape.png filter=lfs diff=lfs merge=lfs -text
38
+ samples/orrery.png filter=lfs diff=lfs merge=lfs -text
39
+ samples/portrait.png filter=lfs diff=lfs merge=lfs -text
40
+ samples/text-board.png filter=lfs diff=lfs merge=lfs -text
41
+ samples/text-sign.png filter=lfs diff=lfs merge=lfs -text
NOTICE ADDED
@@ -0,0 +1 @@
 
 
1
+ Qwen is licensed under the Qwen RESEARCH LICENSE AGREEMENT, Copyright (c) 2026 Hangzhou Tongyi Laboratory Technology Co., Ltd. All Rights Reserved.
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: qwen-research
4
+ license_link: https://huggingface.co/Qwen/Qwen-Image-2.1/blob/main/LICENSE
5
+ base_model: Qwen/Qwen-Image-2.1
6
+ tags: [text-to-image, image-to-image, quantization, w4a4, svdquant, gptq, nunchaku]
7
+ pipeline_tag: text-to-image
8
+ ---
9
+
10
+ # Qwen-Image-2.1 W4A4 (int4)
11
+
12
+ Built with Qwen.
13
+
14
+ > **Non-commercial use only.** An educational artifact under the Qwen Research License,
15
+ > which the base model carries. Commercial use needs a separate licence from the rights
16
+ > holder; publishing this here does not grant one.
17
+
18
+ 4-bit weights, 4-bit activations, for the Qwen-Image-2.1 transformer. Calibrated with
19
+ SVDQuant, error-compensated with GPTQ, packed for Nunchaku's fused kernel.
20
+ **4.34 GB** against about 10 GB for the same layers in bf16.
21
+
22
+ The low-rank branch is fused into the 4-bit GEMM, which is where the speed comes from —
23
+ an unfused branch gives back 24–34% of it.
24
+
25
+ ## Hardware
26
+
27
+ **sm_75 through sm_120** — Ada (4090, L40S) and Blackwell alike.
28
+
29
+ ## Measured
30
+
31
+ | card | cap | seconds | vs bf16 | resident GB | LPIPS vs bf16 |
32
+ |---|---|---|---|---|---|
33
+ | RTX 4090 48GB | sm_89 | 11.576 | 1.84x | 21.35 | 0.231 |
34
+ | RTX PRO 6000 Max-Q | sm_120 | 37.616 | 0.35x | 21.29 | 0.1836 |
35
+
36
+ 40 steps, 1024×1024, seed 42, six prompts. LPIPS is AlexNet against the bf16 image of
37
+ the same prompt on the same card.
38
+
39
+ **GPTQ is worth 6% here**:
40
+ 0.1946 with round-to-nearest against **0.1836** with GPTQ, at the same
41
+ speed, the same file size and the same resident memory — it only changes which grid point
42
+ each weight lands on.
43
+
44
+ Read LPIPS as distance from the bf16 trajectory, not as damage. On a dense-text prompt
45
+ every build renders the text correctly while scoring 0.24–0.35, and on one prompt the
46
+ bf16 reference garbles a sign that this checkpoint renders correctly — and is scored
47
+ worse for it.
48
+
49
+ Only the transformer is quantized. The Qwen3-VL text encoder stays bf16 and is about
50
+ 16.3 GB of a 21.5 GB resident total, so it, not this file, decides how many reference
51
+ images fit on a card.
52
+
53
+ ## Requirements
54
+
55
+ Packed for a specific kernel. The layout is an MMA fragment swizzle, so a different
56
+ Nunchaku build may read it incorrectly — and would produce a wrong image rather than an
57
+ error. The loader checks and warns.
58
+
59
+ ```
60
+ torch==2.12.1
61
+ nunchaku==1.3.0.dev20260306+cu13.0torch2.12
62
+ ```
63
+
64
+ ## Loading
65
+
66
+ ```python
67
+ import torch
68
+ from diffusers import QwenImage21Pipeline
69
+ from quantize import nunchaku_io
70
+
71
+ transformer = nunchaku_io.load_checkpoint(precision="int4", device="cuda")
72
+ pipe = QwenImage21Pipeline.from_pretrained(
73
+ "Qwen/Qwen-Image-2.1", torch_dtype=torch.bfloat16, transformer=transformer)
74
+ ```
75
+
76
+ `nunchaku_io` and the build pipeline: https://github.com/ModelsLab/qwen-image-2-1-server
77
+
78
+ ## How it was built
79
+
80
+ Smoothing factors from real activation statistics (SmoothQuant, α=0.5), a rank-128 SVD
81
+ branch kept in bf16, and the residual rounded onto Nunchaku's grid by GPTQ against a
82
+ calibration Hessian. One Hessian pass, 128 distinct inputs across 224 layers, 24 GB
83
+ resident, ~5 minutes.
84
+
85
+ `w4a4_build.json` records the whole build, including per-layer GPTQ statistics.
86
+
87
+ ## License
88
+
89
+ Derivative of [Qwen/Qwen-Image-2.1](https://huggingface.co/Qwen/Qwen-Image-2.1) under the
90
+ **Qwen Research License Agreement** — non-commercial only. See `NOTICE`.
config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "QwenImage21Transformer2DModel",
3
+ "_diffusers_version": "0.41.0.dev0",
4
+ "_name_or_path": "/workspace/hf/hub/models--Qwen--Qwen-Image-2.1/snapshots/790c92633540aa0cb11d9abf19eb46d861714758/transformer",
5
+ "attention_head_dim": 128,
6
+ "axes_dims_rope": [
7
+ 16,
8
+ 56,
9
+ 56
10
+ ],
11
+ "causal_condition": true,
12
+ "context_in_dim": 4096,
13
+ "eps": 1e-06,
14
+ "in_channels": 64,
15
+ "mlp_ratio": 3,
16
+ "num_attention_heads": 32,
17
+ "num_layers": 32,
18
+ "out_channels": 64,
19
+ "patch_size": 1
20
+ }
diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:966cb6114fb373c2819d2a159e80e3a39c3532ed948564d693a0780d86a07438
3
+ size 4655346432
nunchaku_w4a4.json ADDED
@@ -0,0 +1,1673 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "Qwen/Qwen-Image-2.1",
3
+ "quantization": "nunchaku-svdquant-w4a4",
4
+ "precision": "int4",
5
+ "group_size": 64,
6
+ "rank": 128,
7
+ "layers": {
8
+ "transformer_blocks.0.attn.to_q": {
9
+ "in_features": 4096,
10
+ "out_features": 4096,
11
+ "rank": 128,
12
+ "precision": "int4",
13
+ "bias": false
14
+ },
15
+ "transformer_blocks.0.attn.to_k": {
16
+ "in_features": 4096,
17
+ "out_features": 4096,
18
+ "rank": 128,
19
+ "precision": "int4",
20
+ "bias": false
21
+ },
22
+ "transformer_blocks.0.attn.to_v": {
23
+ "in_features": 4096,
24
+ "out_features": 4096,
25
+ "rank": 128,
26
+ "precision": "int4",
27
+ "bias": false
28
+ },
29
+ "transformer_blocks.0.attn.to_out.0": {
30
+ "in_features": 4096,
31
+ "out_features": 4096,
32
+ "rank": 128,
33
+ "precision": "int4",
34
+ "bias": false
35
+ },
36
+ "transformer_blocks.0.img_mlp.proj": {
37
+ "in_features": 4096,
38
+ "out_features": 12288,
39
+ "rank": 128,
40
+ "precision": "int4",
41
+ "bias": false
42
+ },
43
+ "transformer_blocks.0.img_mlp.out": {
44
+ "in_features": 12288,
45
+ "out_features": 4096,
46
+ "rank": 128,
47
+ "precision": "int4",
48
+ "bias": false
49
+ },
50
+ "transformer_blocks.0.img_mlp.gate_layer": {
51
+ "in_features": 4096,
52
+ "out_features": 12288,
53
+ "rank": 128,
54
+ "precision": "int4",
55
+ "bias": false
56
+ },
57
+ "transformer_blocks.1.attn.to_q": {
58
+ "in_features": 4096,
59
+ "out_features": 4096,
60
+ "rank": 128,
61
+ "precision": "int4",
62
+ "bias": false
63
+ },
64
+ "transformer_blocks.1.attn.to_k": {
65
+ "in_features": 4096,
66
+ "out_features": 4096,
67
+ "rank": 128,
68
+ "precision": "int4",
69
+ "bias": false
70
+ },
71
+ "transformer_blocks.1.attn.to_v": {
72
+ "in_features": 4096,
73
+ "out_features": 4096,
74
+ "rank": 128,
75
+ "precision": "int4",
76
+ "bias": false
77
+ },
78
+ "transformer_blocks.1.attn.to_out.0": {
79
+ "in_features": 4096,
80
+ "out_features": 4096,
81
+ "rank": 128,
82
+ "precision": "int4",
83
+ "bias": false
84
+ },
85
+ "transformer_blocks.1.img_mlp.proj": {
86
+ "in_features": 4096,
87
+ "out_features": 12288,
88
+ "rank": 128,
89
+ "precision": "int4",
90
+ "bias": false
91
+ },
92
+ "transformer_blocks.1.img_mlp.out": {
93
+ "in_features": 12288,
94
+ "out_features": 4096,
95
+ "rank": 128,
96
+ "precision": "int4",
97
+ "bias": false
98
+ },
99
+ "transformer_blocks.1.img_mlp.gate_layer": {
100
+ "in_features": 4096,
101
+ "out_features": 12288,
102
+ "rank": 128,
103
+ "precision": "int4",
104
+ "bias": false
105
+ },
106
+ "transformer_blocks.2.attn.to_q": {
107
+ "in_features": 4096,
108
+ "out_features": 4096,
109
+ "rank": 128,
110
+ "precision": "int4",
111
+ "bias": false
112
+ },
113
+ "transformer_blocks.2.attn.to_k": {
114
+ "in_features": 4096,
115
+ "out_features": 4096,
116
+ "rank": 128,
117
+ "precision": "int4",
118
+ "bias": false
119
+ },
120
+ "transformer_blocks.2.attn.to_v": {
121
+ "in_features": 4096,
122
+ "out_features": 4096,
123
+ "rank": 128,
124
+ "precision": "int4",
125
+ "bias": false
126
+ },
127
+ "transformer_blocks.2.attn.to_out.0": {
128
+ "in_features": 4096,
129
+ "out_features": 4096,
130
+ "rank": 128,
131
+ "precision": "int4",
132
+ "bias": false
133
+ },
134
+ "transformer_blocks.2.img_mlp.proj": {
135
+ "in_features": 4096,
136
+ "out_features": 12288,
137
+ "rank": 128,
138
+ "precision": "int4",
139
+ "bias": false
140
+ },
141
+ "transformer_blocks.2.img_mlp.out": {
142
+ "in_features": 12288,
143
+ "out_features": 4096,
144
+ "rank": 128,
145
+ "precision": "int4",
146
+ "bias": false
147
+ },
148
+ "transformer_blocks.2.img_mlp.gate_layer": {
149
+ "in_features": 4096,
150
+ "out_features": 12288,
151
+ "rank": 128,
152
+ "precision": "int4",
153
+ "bias": false
154
+ },
155
+ "transformer_blocks.3.attn.to_q": {
156
+ "in_features": 4096,
157
+ "out_features": 4096,
158
+ "rank": 128,
159
+ "precision": "int4",
160
+ "bias": false
161
+ },
162
+ "transformer_blocks.3.attn.to_k": {
163
+ "in_features": 4096,
164
+ "out_features": 4096,
165
+ "rank": 128,
166
+ "precision": "int4",
167
+ "bias": false
168
+ },
169
+ "transformer_blocks.3.attn.to_v": {
170
+ "in_features": 4096,
171
+ "out_features": 4096,
172
+ "rank": 128,
173
+ "precision": "int4",
174
+ "bias": false
175
+ },
176
+ "transformer_blocks.3.attn.to_out.0": {
177
+ "in_features": 4096,
178
+ "out_features": 4096,
179
+ "rank": 128,
180
+ "precision": "int4",
181
+ "bias": false
182
+ },
183
+ "transformer_blocks.3.img_mlp.proj": {
184
+ "in_features": 4096,
185
+ "out_features": 12288,
186
+ "rank": 128,
187
+ "precision": "int4",
188
+ "bias": false
189
+ },
190
+ "transformer_blocks.3.img_mlp.out": {
191
+ "in_features": 12288,
192
+ "out_features": 4096,
193
+ "rank": 128,
194
+ "precision": "int4",
195
+ "bias": false
196
+ },
197
+ "transformer_blocks.3.img_mlp.gate_layer": {
198
+ "in_features": 4096,
199
+ "out_features": 12288,
200
+ "rank": 128,
201
+ "precision": "int4",
202
+ "bias": false
203
+ },
204
+ "transformer_blocks.4.attn.to_q": {
205
+ "in_features": 4096,
206
+ "out_features": 4096,
207
+ "rank": 128,
208
+ "precision": "int4",
209
+ "bias": false
210
+ },
211
+ "transformer_blocks.4.attn.to_k": {
212
+ "in_features": 4096,
213
+ "out_features": 4096,
214
+ "rank": 128,
215
+ "precision": "int4",
216
+ "bias": false
217
+ },
218
+ "transformer_blocks.4.attn.to_v": {
219
+ "in_features": 4096,
220
+ "out_features": 4096,
221
+ "rank": 128,
222
+ "precision": "int4",
223
+ "bias": false
224
+ },
225
+ "transformer_blocks.4.attn.to_out.0": {
226
+ "in_features": 4096,
227
+ "out_features": 4096,
228
+ "rank": 128,
229
+ "precision": "int4",
230
+ "bias": false
231
+ },
232
+ "transformer_blocks.4.img_mlp.proj": {
233
+ "in_features": 4096,
234
+ "out_features": 12288,
235
+ "rank": 128,
236
+ "precision": "int4",
237
+ "bias": false
238
+ },
239
+ "transformer_blocks.4.img_mlp.out": {
240
+ "in_features": 12288,
241
+ "out_features": 4096,
242
+ "rank": 128,
243
+ "precision": "int4",
244
+ "bias": false
245
+ },
246
+ "transformer_blocks.4.img_mlp.gate_layer": {
247
+ "in_features": 4096,
248
+ "out_features": 12288,
249
+ "rank": 128,
250
+ "precision": "int4",
251
+ "bias": false
252
+ },
253
+ "transformer_blocks.5.attn.to_q": {
254
+ "in_features": 4096,
255
+ "out_features": 4096,
256
+ "rank": 128,
257
+ "precision": "int4",
258
+ "bias": false
259
+ },
260
+ "transformer_blocks.5.attn.to_k": {
261
+ "in_features": 4096,
262
+ "out_features": 4096,
263
+ "rank": 128,
264
+ "precision": "int4",
265
+ "bias": false
266
+ },
267
+ "transformer_blocks.5.attn.to_v": {
268
+ "in_features": 4096,
269
+ "out_features": 4096,
270
+ "rank": 128,
271
+ "precision": "int4",
272
+ "bias": false
273
+ },
274
+ "transformer_blocks.5.attn.to_out.0": {
275
+ "in_features": 4096,
276
+ "out_features": 4096,
277
+ "rank": 128,
278
+ "precision": "int4",
279
+ "bias": false
280
+ },
281
+ "transformer_blocks.5.img_mlp.proj": {
282
+ "in_features": 4096,
283
+ "out_features": 12288,
284
+ "rank": 128,
285
+ "precision": "int4",
286
+ "bias": false
287
+ },
288
+ "transformer_blocks.5.img_mlp.out": {
289
+ "in_features": 12288,
290
+ "out_features": 4096,
291
+ "rank": 128,
292
+ "precision": "int4",
293
+ "bias": false
294
+ },
295
+ "transformer_blocks.5.img_mlp.gate_layer": {
296
+ "in_features": 4096,
297
+ "out_features": 12288,
298
+ "rank": 128,
299
+ "precision": "int4",
300
+ "bias": false
301
+ },
302
+ "transformer_blocks.6.attn.to_q": {
303
+ "in_features": 4096,
304
+ "out_features": 4096,
305
+ "rank": 128,
306
+ "precision": "int4",
307
+ "bias": false
308
+ },
309
+ "transformer_blocks.6.attn.to_k": {
310
+ "in_features": 4096,
311
+ "out_features": 4096,
312
+ "rank": 128,
313
+ "precision": "int4",
314
+ "bias": false
315
+ },
316
+ "transformer_blocks.6.attn.to_v": {
317
+ "in_features": 4096,
318
+ "out_features": 4096,
319
+ "rank": 128,
320
+ "precision": "int4",
321
+ "bias": false
322
+ },
323
+ "transformer_blocks.6.attn.to_out.0": {
324
+ "in_features": 4096,
325
+ "out_features": 4096,
326
+ "rank": 128,
327
+ "precision": "int4",
328
+ "bias": false
329
+ },
330
+ "transformer_blocks.6.img_mlp.proj": {
331
+ "in_features": 4096,
332
+ "out_features": 12288,
333
+ "rank": 128,
334
+ "precision": "int4",
335
+ "bias": false
336
+ },
337
+ "transformer_blocks.6.img_mlp.out": {
338
+ "in_features": 12288,
339
+ "out_features": 4096,
340
+ "rank": 128,
341
+ "precision": "int4",
342
+ "bias": false
343
+ },
344
+ "transformer_blocks.6.img_mlp.gate_layer": {
345
+ "in_features": 4096,
346
+ "out_features": 12288,
347
+ "rank": 128,
348
+ "precision": "int4",
349
+ "bias": false
350
+ },
351
+ "transformer_blocks.7.attn.to_q": {
352
+ "in_features": 4096,
353
+ "out_features": 4096,
354
+ "rank": 128,
355
+ "precision": "int4",
356
+ "bias": false
357
+ },
358
+ "transformer_blocks.7.attn.to_k": {
359
+ "in_features": 4096,
360
+ "out_features": 4096,
361
+ "rank": 128,
362
+ "precision": "int4",
363
+ "bias": false
364
+ },
365
+ "transformer_blocks.7.attn.to_v": {
366
+ "in_features": 4096,
367
+ "out_features": 4096,
368
+ "rank": 128,
369
+ "precision": "int4",
370
+ "bias": false
371
+ },
372
+ "transformer_blocks.7.attn.to_out.0": {
373
+ "in_features": 4096,
374
+ "out_features": 4096,
375
+ "rank": 128,
376
+ "precision": "int4",
377
+ "bias": false
378
+ },
379
+ "transformer_blocks.7.img_mlp.proj": {
380
+ "in_features": 4096,
381
+ "out_features": 12288,
382
+ "rank": 128,
383
+ "precision": "int4",
384
+ "bias": false
385
+ },
386
+ "transformer_blocks.7.img_mlp.out": {
387
+ "in_features": 12288,
388
+ "out_features": 4096,
389
+ "rank": 128,
390
+ "precision": "int4",
391
+ "bias": false
392
+ },
393
+ "transformer_blocks.7.img_mlp.gate_layer": {
394
+ "in_features": 4096,
395
+ "out_features": 12288,
396
+ "rank": 128,
397
+ "precision": "int4",
398
+ "bias": false
399
+ },
400
+ "transformer_blocks.8.attn.to_q": {
401
+ "in_features": 4096,
402
+ "out_features": 4096,
403
+ "rank": 128,
404
+ "precision": "int4",
405
+ "bias": false
406
+ },
407
+ "transformer_blocks.8.attn.to_k": {
408
+ "in_features": 4096,
409
+ "out_features": 4096,
410
+ "rank": 128,
411
+ "precision": "int4",
412
+ "bias": false
413
+ },
414
+ "transformer_blocks.8.attn.to_v": {
415
+ "in_features": 4096,
416
+ "out_features": 4096,
417
+ "rank": 128,
418
+ "precision": "int4",
419
+ "bias": false
420
+ },
421
+ "transformer_blocks.8.attn.to_out.0": {
422
+ "in_features": 4096,
423
+ "out_features": 4096,
424
+ "rank": 128,
425
+ "precision": "int4",
426
+ "bias": false
427
+ },
428
+ "transformer_blocks.8.img_mlp.proj": {
429
+ "in_features": 4096,
430
+ "out_features": 12288,
431
+ "rank": 128,
432
+ "precision": "int4",
433
+ "bias": false
434
+ },
435
+ "transformer_blocks.8.img_mlp.out": {
436
+ "in_features": 12288,
437
+ "out_features": 4096,
438
+ "rank": 128,
439
+ "precision": "int4",
440
+ "bias": false
441
+ },
442
+ "transformer_blocks.8.img_mlp.gate_layer": {
443
+ "in_features": 4096,
444
+ "out_features": 12288,
445
+ "rank": 128,
446
+ "precision": "int4",
447
+ "bias": false
448
+ },
449
+ "transformer_blocks.9.attn.to_q": {
450
+ "in_features": 4096,
451
+ "out_features": 4096,
452
+ "rank": 128,
453
+ "precision": "int4",
454
+ "bias": false
455
+ },
456
+ "transformer_blocks.9.attn.to_k": {
457
+ "in_features": 4096,
458
+ "out_features": 4096,
459
+ "rank": 128,
460
+ "precision": "int4",
461
+ "bias": false
462
+ },
463
+ "transformer_blocks.9.attn.to_v": {
464
+ "in_features": 4096,
465
+ "out_features": 4096,
466
+ "rank": 128,
467
+ "precision": "int4",
468
+ "bias": false
469
+ },
470
+ "transformer_blocks.9.attn.to_out.0": {
471
+ "in_features": 4096,
472
+ "out_features": 4096,
473
+ "rank": 128,
474
+ "precision": "int4",
475
+ "bias": false
476
+ },
477
+ "transformer_blocks.9.img_mlp.proj": {
478
+ "in_features": 4096,
479
+ "out_features": 12288,
480
+ "rank": 128,
481
+ "precision": "int4",
482
+ "bias": false
483
+ },
484
+ "transformer_blocks.9.img_mlp.out": {
485
+ "in_features": 12288,
486
+ "out_features": 4096,
487
+ "rank": 128,
488
+ "precision": "int4",
489
+ "bias": false
490
+ },
491
+ "transformer_blocks.9.img_mlp.gate_layer": {
492
+ "in_features": 4096,
493
+ "out_features": 12288,
494
+ "rank": 128,
495
+ "precision": "int4",
496
+ "bias": false
497
+ },
498
+ "transformer_blocks.10.attn.to_q": {
499
+ "in_features": 4096,
500
+ "out_features": 4096,
501
+ "rank": 128,
502
+ "precision": "int4",
503
+ "bias": false
504
+ },
505
+ "transformer_blocks.10.attn.to_k": {
506
+ "in_features": 4096,
507
+ "out_features": 4096,
508
+ "rank": 128,
509
+ "precision": "int4",
510
+ "bias": false
511
+ },
512
+ "transformer_blocks.10.attn.to_v": {
513
+ "in_features": 4096,
514
+ "out_features": 4096,
515
+ "rank": 128,
516
+ "precision": "int4",
517
+ "bias": false
518
+ },
519
+ "transformer_blocks.10.attn.to_out.0": {
520
+ "in_features": 4096,
521
+ "out_features": 4096,
522
+ "rank": 128,
523
+ "precision": "int4",
524
+ "bias": false
525
+ },
526
+ "transformer_blocks.10.img_mlp.proj": {
527
+ "in_features": 4096,
528
+ "out_features": 12288,
529
+ "rank": 128,
530
+ "precision": "int4",
531
+ "bias": false
532
+ },
533
+ "transformer_blocks.10.img_mlp.out": {
534
+ "in_features": 12288,
535
+ "out_features": 4096,
536
+ "rank": 128,
537
+ "precision": "int4",
538
+ "bias": false
539
+ },
540
+ "transformer_blocks.10.img_mlp.gate_layer": {
541
+ "in_features": 4096,
542
+ "out_features": 12288,
543
+ "rank": 128,
544
+ "precision": "int4",
545
+ "bias": false
546
+ },
547
+ "transformer_blocks.11.attn.to_q": {
548
+ "in_features": 4096,
549
+ "out_features": 4096,
550
+ "rank": 128,
551
+ "precision": "int4",
552
+ "bias": false
553
+ },
554
+ "transformer_blocks.11.attn.to_k": {
555
+ "in_features": 4096,
556
+ "out_features": 4096,
557
+ "rank": 128,
558
+ "precision": "int4",
559
+ "bias": false
560
+ },
561
+ "transformer_blocks.11.attn.to_v": {
562
+ "in_features": 4096,
563
+ "out_features": 4096,
564
+ "rank": 128,
565
+ "precision": "int4",
566
+ "bias": false
567
+ },
568
+ "transformer_blocks.11.attn.to_out.0": {
569
+ "in_features": 4096,
570
+ "out_features": 4096,
571
+ "rank": 128,
572
+ "precision": "int4",
573
+ "bias": false
574
+ },
575
+ "transformer_blocks.11.img_mlp.proj": {
576
+ "in_features": 4096,
577
+ "out_features": 12288,
578
+ "rank": 128,
579
+ "precision": "int4",
580
+ "bias": false
581
+ },
582
+ "transformer_blocks.11.img_mlp.out": {
583
+ "in_features": 12288,
584
+ "out_features": 4096,
585
+ "rank": 128,
586
+ "precision": "int4",
587
+ "bias": false
588
+ },
589
+ "transformer_blocks.11.img_mlp.gate_layer": {
590
+ "in_features": 4096,
591
+ "out_features": 12288,
592
+ "rank": 128,
593
+ "precision": "int4",
594
+ "bias": false
595
+ },
596
+ "transformer_blocks.12.attn.to_q": {
597
+ "in_features": 4096,
598
+ "out_features": 4096,
599
+ "rank": 128,
600
+ "precision": "int4",
601
+ "bias": false
602
+ },
603
+ "transformer_blocks.12.attn.to_k": {
604
+ "in_features": 4096,
605
+ "out_features": 4096,
606
+ "rank": 128,
607
+ "precision": "int4",
608
+ "bias": false
609
+ },
610
+ "transformer_blocks.12.attn.to_v": {
611
+ "in_features": 4096,
612
+ "out_features": 4096,
613
+ "rank": 128,
614
+ "precision": "int4",
615
+ "bias": false
616
+ },
617
+ "transformer_blocks.12.attn.to_out.0": {
618
+ "in_features": 4096,
619
+ "out_features": 4096,
620
+ "rank": 128,
621
+ "precision": "int4",
622
+ "bias": false
623
+ },
624
+ "transformer_blocks.12.img_mlp.proj": {
625
+ "in_features": 4096,
626
+ "out_features": 12288,
627
+ "rank": 128,
628
+ "precision": "int4",
629
+ "bias": false
630
+ },
631
+ "transformer_blocks.12.img_mlp.out": {
632
+ "in_features": 12288,
633
+ "out_features": 4096,
634
+ "rank": 128,
635
+ "precision": "int4",
636
+ "bias": false
637
+ },
638
+ "transformer_blocks.12.img_mlp.gate_layer": {
639
+ "in_features": 4096,
640
+ "out_features": 12288,
641
+ "rank": 128,
642
+ "precision": "int4",
643
+ "bias": false
644
+ },
645
+ "transformer_blocks.13.attn.to_q": {
646
+ "in_features": 4096,
647
+ "out_features": 4096,
648
+ "rank": 128,
649
+ "precision": "int4",
650
+ "bias": false
651
+ },
652
+ "transformer_blocks.13.attn.to_k": {
653
+ "in_features": 4096,
654
+ "out_features": 4096,
655
+ "rank": 128,
656
+ "precision": "int4",
657
+ "bias": false
658
+ },
659
+ "transformer_blocks.13.attn.to_v": {
660
+ "in_features": 4096,
661
+ "out_features": 4096,
662
+ "rank": 128,
663
+ "precision": "int4",
664
+ "bias": false
665
+ },
666
+ "transformer_blocks.13.attn.to_out.0": {
667
+ "in_features": 4096,
668
+ "out_features": 4096,
669
+ "rank": 128,
670
+ "precision": "int4",
671
+ "bias": false
672
+ },
673
+ "transformer_blocks.13.img_mlp.proj": {
674
+ "in_features": 4096,
675
+ "out_features": 12288,
676
+ "rank": 128,
677
+ "precision": "int4",
678
+ "bias": false
679
+ },
680
+ "transformer_blocks.13.img_mlp.out": {
681
+ "in_features": 12288,
682
+ "out_features": 4096,
683
+ "rank": 128,
684
+ "precision": "int4",
685
+ "bias": false
686
+ },
687
+ "transformer_blocks.13.img_mlp.gate_layer": {
688
+ "in_features": 4096,
689
+ "out_features": 12288,
690
+ "rank": 128,
691
+ "precision": "int4",
692
+ "bias": false
693
+ },
694
+ "transformer_blocks.14.attn.to_q": {
695
+ "in_features": 4096,
696
+ "out_features": 4096,
697
+ "rank": 128,
698
+ "precision": "int4",
699
+ "bias": false
700
+ },
701
+ "transformer_blocks.14.attn.to_k": {
702
+ "in_features": 4096,
703
+ "out_features": 4096,
704
+ "rank": 128,
705
+ "precision": "int4",
706
+ "bias": false
707
+ },
708
+ "transformer_blocks.14.attn.to_v": {
709
+ "in_features": 4096,
710
+ "out_features": 4096,
711
+ "rank": 128,
712
+ "precision": "int4",
713
+ "bias": false
714
+ },
715
+ "transformer_blocks.14.attn.to_out.0": {
716
+ "in_features": 4096,
717
+ "out_features": 4096,
718
+ "rank": 128,
719
+ "precision": "int4",
720
+ "bias": false
721
+ },
722
+ "transformer_blocks.14.img_mlp.proj": {
723
+ "in_features": 4096,
724
+ "out_features": 12288,
725
+ "rank": 128,
726
+ "precision": "int4",
727
+ "bias": false
728
+ },
729
+ "transformer_blocks.14.img_mlp.out": {
730
+ "in_features": 12288,
731
+ "out_features": 4096,
732
+ "rank": 128,
733
+ "precision": "int4",
734
+ "bias": false
735
+ },
736
+ "transformer_blocks.14.img_mlp.gate_layer": {
737
+ "in_features": 4096,
738
+ "out_features": 12288,
739
+ "rank": 128,
740
+ "precision": "int4",
741
+ "bias": false
742
+ },
743
+ "transformer_blocks.15.attn.to_q": {
744
+ "in_features": 4096,
745
+ "out_features": 4096,
746
+ "rank": 128,
747
+ "precision": "int4",
748
+ "bias": false
749
+ },
750
+ "transformer_blocks.15.attn.to_k": {
751
+ "in_features": 4096,
752
+ "out_features": 4096,
753
+ "rank": 128,
754
+ "precision": "int4",
755
+ "bias": false
756
+ },
757
+ "transformer_blocks.15.attn.to_v": {
758
+ "in_features": 4096,
759
+ "out_features": 4096,
760
+ "rank": 128,
761
+ "precision": "int4",
762
+ "bias": false
763
+ },
764
+ "transformer_blocks.15.attn.to_out.0": {
765
+ "in_features": 4096,
766
+ "out_features": 4096,
767
+ "rank": 128,
768
+ "precision": "int4",
769
+ "bias": false
770
+ },
771
+ "transformer_blocks.15.img_mlp.proj": {
772
+ "in_features": 4096,
773
+ "out_features": 12288,
774
+ "rank": 128,
775
+ "precision": "int4",
776
+ "bias": false
777
+ },
778
+ "transformer_blocks.15.img_mlp.out": {
779
+ "in_features": 12288,
780
+ "out_features": 4096,
781
+ "rank": 128,
782
+ "precision": "int4",
783
+ "bias": false
784
+ },
785
+ "transformer_blocks.15.img_mlp.gate_layer": {
786
+ "in_features": 4096,
787
+ "out_features": 12288,
788
+ "rank": 128,
789
+ "precision": "int4",
790
+ "bias": false
791
+ },
792
+ "transformer_blocks.16.attn.to_q": {
793
+ "in_features": 4096,
794
+ "out_features": 4096,
795
+ "rank": 128,
796
+ "precision": "int4",
797
+ "bias": false
798
+ },
799
+ "transformer_blocks.16.attn.to_k": {
800
+ "in_features": 4096,
801
+ "out_features": 4096,
802
+ "rank": 128,
803
+ "precision": "int4",
804
+ "bias": false
805
+ },
806
+ "transformer_blocks.16.attn.to_v": {
807
+ "in_features": 4096,
808
+ "out_features": 4096,
809
+ "rank": 128,
810
+ "precision": "int4",
811
+ "bias": false
812
+ },
813
+ "transformer_blocks.16.attn.to_out.0": {
814
+ "in_features": 4096,
815
+ "out_features": 4096,
816
+ "rank": 128,
817
+ "precision": "int4",
818
+ "bias": false
819
+ },
820
+ "transformer_blocks.16.img_mlp.proj": {
821
+ "in_features": 4096,
822
+ "out_features": 12288,
823
+ "rank": 128,
824
+ "precision": "int4",
825
+ "bias": false
826
+ },
827
+ "transformer_blocks.16.img_mlp.out": {
828
+ "in_features": 12288,
829
+ "out_features": 4096,
830
+ "rank": 128,
831
+ "precision": "int4",
832
+ "bias": false
833
+ },
834
+ "transformer_blocks.16.img_mlp.gate_layer": {
835
+ "in_features": 4096,
836
+ "out_features": 12288,
837
+ "rank": 128,
838
+ "precision": "int4",
839
+ "bias": false
840
+ },
841
+ "transformer_blocks.17.attn.to_q": {
842
+ "in_features": 4096,
843
+ "out_features": 4096,
844
+ "rank": 128,
845
+ "precision": "int4",
846
+ "bias": false
847
+ },
848
+ "transformer_blocks.17.attn.to_k": {
849
+ "in_features": 4096,
850
+ "out_features": 4096,
851
+ "rank": 128,
852
+ "precision": "int4",
853
+ "bias": false
854
+ },
855
+ "transformer_blocks.17.attn.to_v": {
856
+ "in_features": 4096,
857
+ "out_features": 4096,
858
+ "rank": 128,
859
+ "precision": "int4",
860
+ "bias": false
861
+ },
862
+ "transformer_blocks.17.attn.to_out.0": {
863
+ "in_features": 4096,
864
+ "out_features": 4096,
865
+ "rank": 128,
866
+ "precision": "int4",
867
+ "bias": false
868
+ },
869
+ "transformer_blocks.17.img_mlp.proj": {
870
+ "in_features": 4096,
871
+ "out_features": 12288,
872
+ "rank": 128,
873
+ "precision": "int4",
874
+ "bias": false
875
+ },
876
+ "transformer_blocks.17.img_mlp.out": {
877
+ "in_features": 12288,
878
+ "out_features": 4096,
879
+ "rank": 128,
880
+ "precision": "int4",
881
+ "bias": false
882
+ },
883
+ "transformer_blocks.17.img_mlp.gate_layer": {
884
+ "in_features": 4096,
885
+ "out_features": 12288,
886
+ "rank": 128,
887
+ "precision": "int4",
888
+ "bias": false
889
+ },
890
+ "transformer_blocks.18.attn.to_q": {
891
+ "in_features": 4096,
892
+ "out_features": 4096,
893
+ "rank": 128,
894
+ "precision": "int4",
895
+ "bias": false
896
+ },
897
+ "transformer_blocks.18.attn.to_k": {
898
+ "in_features": 4096,
899
+ "out_features": 4096,
900
+ "rank": 128,
901
+ "precision": "int4",
902
+ "bias": false
903
+ },
904
+ "transformer_blocks.18.attn.to_v": {
905
+ "in_features": 4096,
906
+ "out_features": 4096,
907
+ "rank": 128,
908
+ "precision": "int4",
909
+ "bias": false
910
+ },
911
+ "transformer_blocks.18.attn.to_out.0": {
912
+ "in_features": 4096,
913
+ "out_features": 4096,
914
+ "rank": 128,
915
+ "precision": "int4",
916
+ "bias": false
917
+ },
918
+ "transformer_blocks.18.img_mlp.proj": {
919
+ "in_features": 4096,
920
+ "out_features": 12288,
921
+ "rank": 128,
922
+ "precision": "int4",
923
+ "bias": false
924
+ },
925
+ "transformer_blocks.18.img_mlp.out": {
926
+ "in_features": 12288,
927
+ "out_features": 4096,
928
+ "rank": 128,
929
+ "precision": "int4",
930
+ "bias": false
931
+ },
932
+ "transformer_blocks.18.img_mlp.gate_layer": {
933
+ "in_features": 4096,
934
+ "out_features": 12288,
935
+ "rank": 128,
936
+ "precision": "int4",
937
+ "bias": false
938
+ },
939
+ "transformer_blocks.19.attn.to_q": {
940
+ "in_features": 4096,
941
+ "out_features": 4096,
942
+ "rank": 128,
943
+ "precision": "int4",
944
+ "bias": false
945
+ },
946
+ "transformer_blocks.19.attn.to_k": {
947
+ "in_features": 4096,
948
+ "out_features": 4096,
949
+ "rank": 128,
950
+ "precision": "int4",
951
+ "bias": false
952
+ },
953
+ "transformer_blocks.19.attn.to_v": {
954
+ "in_features": 4096,
955
+ "out_features": 4096,
956
+ "rank": 128,
957
+ "precision": "int4",
958
+ "bias": false
959
+ },
960
+ "transformer_blocks.19.attn.to_out.0": {
961
+ "in_features": 4096,
962
+ "out_features": 4096,
963
+ "rank": 128,
964
+ "precision": "int4",
965
+ "bias": false
966
+ },
967
+ "transformer_blocks.19.img_mlp.proj": {
968
+ "in_features": 4096,
969
+ "out_features": 12288,
970
+ "rank": 128,
971
+ "precision": "int4",
972
+ "bias": false
973
+ },
974
+ "transformer_blocks.19.img_mlp.out": {
975
+ "in_features": 12288,
976
+ "out_features": 4096,
977
+ "rank": 128,
978
+ "precision": "int4",
979
+ "bias": false
980
+ },
981
+ "transformer_blocks.19.img_mlp.gate_layer": {
982
+ "in_features": 4096,
983
+ "out_features": 12288,
984
+ "rank": 128,
985
+ "precision": "int4",
986
+ "bias": false
987
+ },
988
+ "transformer_blocks.20.attn.to_q": {
989
+ "in_features": 4096,
990
+ "out_features": 4096,
991
+ "rank": 128,
992
+ "precision": "int4",
993
+ "bias": false
994
+ },
995
+ "transformer_blocks.20.attn.to_k": {
996
+ "in_features": 4096,
997
+ "out_features": 4096,
998
+ "rank": 128,
999
+ "precision": "int4",
1000
+ "bias": false
1001
+ },
1002
+ "transformer_blocks.20.attn.to_v": {
1003
+ "in_features": 4096,
1004
+ "out_features": 4096,
1005
+ "rank": 128,
1006
+ "precision": "int4",
1007
+ "bias": false
1008
+ },
1009
+ "transformer_blocks.20.attn.to_out.0": {
1010
+ "in_features": 4096,
1011
+ "out_features": 4096,
1012
+ "rank": 128,
1013
+ "precision": "int4",
1014
+ "bias": false
1015
+ },
1016
+ "transformer_blocks.20.img_mlp.proj": {
1017
+ "in_features": 4096,
1018
+ "out_features": 12288,
1019
+ "rank": 128,
1020
+ "precision": "int4",
1021
+ "bias": false
1022
+ },
1023
+ "transformer_blocks.20.img_mlp.out": {
1024
+ "in_features": 12288,
1025
+ "out_features": 4096,
1026
+ "rank": 128,
1027
+ "precision": "int4",
1028
+ "bias": false
1029
+ },
1030
+ "transformer_blocks.20.img_mlp.gate_layer": {
1031
+ "in_features": 4096,
1032
+ "out_features": 12288,
1033
+ "rank": 128,
1034
+ "precision": "int4",
1035
+ "bias": false
1036
+ },
1037
+ "transformer_blocks.21.attn.to_q": {
1038
+ "in_features": 4096,
1039
+ "out_features": 4096,
1040
+ "rank": 128,
1041
+ "precision": "int4",
1042
+ "bias": false
1043
+ },
1044
+ "transformer_blocks.21.attn.to_k": {
1045
+ "in_features": 4096,
1046
+ "out_features": 4096,
1047
+ "rank": 128,
1048
+ "precision": "int4",
1049
+ "bias": false
1050
+ },
1051
+ "transformer_blocks.21.attn.to_v": {
1052
+ "in_features": 4096,
1053
+ "out_features": 4096,
1054
+ "rank": 128,
1055
+ "precision": "int4",
1056
+ "bias": false
1057
+ },
1058
+ "transformer_blocks.21.attn.to_out.0": {
1059
+ "in_features": 4096,
1060
+ "out_features": 4096,
1061
+ "rank": 128,
1062
+ "precision": "int4",
1063
+ "bias": false
1064
+ },
1065
+ "transformer_blocks.21.img_mlp.proj": {
1066
+ "in_features": 4096,
1067
+ "out_features": 12288,
1068
+ "rank": 128,
1069
+ "precision": "int4",
1070
+ "bias": false
1071
+ },
1072
+ "transformer_blocks.21.img_mlp.out": {
1073
+ "in_features": 12288,
1074
+ "out_features": 4096,
1075
+ "rank": 128,
1076
+ "precision": "int4",
1077
+ "bias": false
1078
+ },
1079
+ "transformer_blocks.21.img_mlp.gate_layer": {
1080
+ "in_features": 4096,
1081
+ "out_features": 12288,
1082
+ "rank": 128,
1083
+ "precision": "int4",
1084
+ "bias": false
1085
+ },
1086
+ "transformer_blocks.22.attn.to_q": {
1087
+ "in_features": 4096,
1088
+ "out_features": 4096,
1089
+ "rank": 128,
1090
+ "precision": "int4",
1091
+ "bias": false
1092
+ },
1093
+ "transformer_blocks.22.attn.to_k": {
1094
+ "in_features": 4096,
1095
+ "out_features": 4096,
1096
+ "rank": 128,
1097
+ "precision": "int4",
1098
+ "bias": false
1099
+ },
1100
+ "transformer_blocks.22.attn.to_v": {
1101
+ "in_features": 4096,
1102
+ "out_features": 4096,
1103
+ "rank": 128,
1104
+ "precision": "int4",
1105
+ "bias": false
1106
+ },
1107
+ "transformer_blocks.22.attn.to_out.0": {
1108
+ "in_features": 4096,
1109
+ "out_features": 4096,
1110
+ "rank": 128,
1111
+ "precision": "int4",
1112
+ "bias": false
1113
+ },
1114
+ "transformer_blocks.22.img_mlp.proj": {
1115
+ "in_features": 4096,
1116
+ "out_features": 12288,
1117
+ "rank": 128,
1118
+ "precision": "int4",
1119
+ "bias": false
1120
+ },
1121
+ "transformer_blocks.22.img_mlp.out": {
1122
+ "in_features": 12288,
1123
+ "out_features": 4096,
1124
+ "rank": 128,
1125
+ "precision": "int4",
1126
+ "bias": false
1127
+ },
1128
+ "transformer_blocks.22.img_mlp.gate_layer": {
1129
+ "in_features": 4096,
1130
+ "out_features": 12288,
1131
+ "rank": 128,
1132
+ "precision": "int4",
1133
+ "bias": false
1134
+ },
1135
+ "transformer_blocks.23.attn.to_q": {
1136
+ "in_features": 4096,
1137
+ "out_features": 4096,
1138
+ "rank": 128,
1139
+ "precision": "int4",
1140
+ "bias": false
1141
+ },
1142
+ "transformer_blocks.23.attn.to_k": {
1143
+ "in_features": 4096,
1144
+ "out_features": 4096,
1145
+ "rank": 128,
1146
+ "precision": "int4",
1147
+ "bias": false
1148
+ },
1149
+ "transformer_blocks.23.attn.to_v": {
1150
+ "in_features": 4096,
1151
+ "out_features": 4096,
1152
+ "rank": 128,
1153
+ "precision": "int4",
1154
+ "bias": false
1155
+ },
1156
+ "transformer_blocks.23.attn.to_out.0": {
1157
+ "in_features": 4096,
1158
+ "out_features": 4096,
1159
+ "rank": 128,
1160
+ "precision": "int4",
1161
+ "bias": false
1162
+ },
1163
+ "transformer_blocks.23.img_mlp.proj": {
1164
+ "in_features": 4096,
1165
+ "out_features": 12288,
1166
+ "rank": 128,
1167
+ "precision": "int4",
1168
+ "bias": false
1169
+ },
1170
+ "transformer_blocks.23.img_mlp.out": {
1171
+ "in_features": 12288,
1172
+ "out_features": 4096,
1173
+ "rank": 128,
1174
+ "precision": "int4",
1175
+ "bias": false
1176
+ },
1177
+ "transformer_blocks.23.img_mlp.gate_layer": {
1178
+ "in_features": 4096,
1179
+ "out_features": 12288,
1180
+ "rank": 128,
1181
+ "precision": "int4",
1182
+ "bias": false
1183
+ },
1184
+ "transformer_blocks.24.attn.to_q": {
1185
+ "in_features": 4096,
1186
+ "out_features": 4096,
1187
+ "rank": 128,
1188
+ "precision": "int4",
1189
+ "bias": false
1190
+ },
1191
+ "transformer_blocks.24.attn.to_k": {
1192
+ "in_features": 4096,
1193
+ "out_features": 4096,
1194
+ "rank": 128,
1195
+ "precision": "int4",
1196
+ "bias": false
1197
+ },
1198
+ "transformer_blocks.24.attn.to_v": {
1199
+ "in_features": 4096,
1200
+ "out_features": 4096,
1201
+ "rank": 128,
1202
+ "precision": "int4",
1203
+ "bias": false
1204
+ },
1205
+ "transformer_blocks.24.attn.to_out.0": {
1206
+ "in_features": 4096,
1207
+ "out_features": 4096,
1208
+ "rank": 128,
1209
+ "precision": "int4",
1210
+ "bias": false
1211
+ },
1212
+ "transformer_blocks.24.img_mlp.proj": {
1213
+ "in_features": 4096,
1214
+ "out_features": 12288,
1215
+ "rank": 128,
1216
+ "precision": "int4",
1217
+ "bias": false
1218
+ },
1219
+ "transformer_blocks.24.img_mlp.out": {
1220
+ "in_features": 12288,
1221
+ "out_features": 4096,
1222
+ "rank": 128,
1223
+ "precision": "int4",
1224
+ "bias": false
1225
+ },
1226
+ "transformer_blocks.24.img_mlp.gate_layer": {
1227
+ "in_features": 4096,
1228
+ "out_features": 12288,
1229
+ "rank": 128,
1230
+ "precision": "int4",
1231
+ "bias": false
1232
+ },
1233
+ "transformer_blocks.25.attn.to_q": {
1234
+ "in_features": 4096,
1235
+ "out_features": 4096,
1236
+ "rank": 128,
1237
+ "precision": "int4",
1238
+ "bias": false
1239
+ },
1240
+ "transformer_blocks.25.attn.to_k": {
1241
+ "in_features": 4096,
1242
+ "out_features": 4096,
1243
+ "rank": 128,
1244
+ "precision": "int4",
1245
+ "bias": false
1246
+ },
1247
+ "transformer_blocks.25.attn.to_v": {
1248
+ "in_features": 4096,
1249
+ "out_features": 4096,
1250
+ "rank": 128,
1251
+ "precision": "int4",
1252
+ "bias": false
1253
+ },
1254
+ "transformer_blocks.25.attn.to_out.0": {
1255
+ "in_features": 4096,
1256
+ "out_features": 4096,
1257
+ "rank": 128,
1258
+ "precision": "int4",
1259
+ "bias": false
1260
+ },
1261
+ "transformer_blocks.25.img_mlp.proj": {
1262
+ "in_features": 4096,
1263
+ "out_features": 12288,
1264
+ "rank": 128,
1265
+ "precision": "int4",
1266
+ "bias": false
1267
+ },
1268
+ "transformer_blocks.25.img_mlp.out": {
1269
+ "in_features": 12288,
1270
+ "out_features": 4096,
1271
+ "rank": 128,
1272
+ "precision": "int4",
1273
+ "bias": false
1274
+ },
1275
+ "transformer_blocks.25.img_mlp.gate_layer": {
1276
+ "in_features": 4096,
1277
+ "out_features": 12288,
1278
+ "rank": 128,
1279
+ "precision": "int4",
1280
+ "bias": false
1281
+ },
1282
+ "transformer_blocks.26.attn.to_q": {
1283
+ "in_features": 4096,
1284
+ "out_features": 4096,
1285
+ "rank": 128,
1286
+ "precision": "int4",
1287
+ "bias": false
1288
+ },
1289
+ "transformer_blocks.26.attn.to_k": {
1290
+ "in_features": 4096,
1291
+ "out_features": 4096,
1292
+ "rank": 128,
1293
+ "precision": "int4",
1294
+ "bias": false
1295
+ },
1296
+ "transformer_blocks.26.attn.to_v": {
1297
+ "in_features": 4096,
1298
+ "out_features": 4096,
1299
+ "rank": 128,
1300
+ "precision": "int4",
1301
+ "bias": false
1302
+ },
1303
+ "transformer_blocks.26.attn.to_out.0": {
1304
+ "in_features": 4096,
1305
+ "out_features": 4096,
1306
+ "rank": 128,
1307
+ "precision": "int4",
1308
+ "bias": false
1309
+ },
1310
+ "transformer_blocks.26.img_mlp.proj": {
1311
+ "in_features": 4096,
1312
+ "out_features": 12288,
1313
+ "rank": 128,
1314
+ "precision": "int4",
1315
+ "bias": false
1316
+ },
1317
+ "transformer_blocks.26.img_mlp.out": {
1318
+ "in_features": 12288,
1319
+ "out_features": 4096,
1320
+ "rank": 128,
1321
+ "precision": "int4",
1322
+ "bias": false
1323
+ },
1324
+ "transformer_blocks.26.img_mlp.gate_layer": {
1325
+ "in_features": 4096,
1326
+ "out_features": 12288,
1327
+ "rank": 128,
1328
+ "precision": "int4",
1329
+ "bias": false
1330
+ },
1331
+ "transformer_blocks.27.attn.to_q": {
1332
+ "in_features": 4096,
1333
+ "out_features": 4096,
1334
+ "rank": 128,
1335
+ "precision": "int4",
1336
+ "bias": false
1337
+ },
1338
+ "transformer_blocks.27.attn.to_k": {
1339
+ "in_features": 4096,
1340
+ "out_features": 4096,
1341
+ "rank": 128,
1342
+ "precision": "int4",
1343
+ "bias": false
1344
+ },
1345
+ "transformer_blocks.27.attn.to_v": {
1346
+ "in_features": 4096,
1347
+ "out_features": 4096,
1348
+ "rank": 128,
1349
+ "precision": "int4",
1350
+ "bias": false
1351
+ },
1352
+ "transformer_blocks.27.attn.to_out.0": {
1353
+ "in_features": 4096,
1354
+ "out_features": 4096,
1355
+ "rank": 128,
1356
+ "precision": "int4",
1357
+ "bias": false
1358
+ },
1359
+ "transformer_blocks.27.img_mlp.proj": {
1360
+ "in_features": 4096,
1361
+ "out_features": 12288,
1362
+ "rank": 128,
1363
+ "precision": "int4",
1364
+ "bias": false
1365
+ },
1366
+ "transformer_blocks.27.img_mlp.out": {
1367
+ "in_features": 12288,
1368
+ "out_features": 4096,
1369
+ "rank": 128,
1370
+ "precision": "int4",
1371
+ "bias": false
1372
+ },
1373
+ "transformer_blocks.27.img_mlp.gate_layer": {
1374
+ "in_features": 4096,
1375
+ "out_features": 12288,
1376
+ "rank": 128,
1377
+ "precision": "int4",
1378
+ "bias": false
1379
+ },
1380
+ "transformer_blocks.28.attn.to_q": {
1381
+ "in_features": 4096,
1382
+ "out_features": 4096,
1383
+ "rank": 128,
1384
+ "precision": "int4",
1385
+ "bias": false
1386
+ },
1387
+ "transformer_blocks.28.attn.to_k": {
1388
+ "in_features": 4096,
1389
+ "out_features": 4096,
1390
+ "rank": 128,
1391
+ "precision": "int4",
1392
+ "bias": false
1393
+ },
1394
+ "transformer_blocks.28.attn.to_v": {
1395
+ "in_features": 4096,
1396
+ "out_features": 4096,
1397
+ "rank": 128,
1398
+ "precision": "int4",
1399
+ "bias": false
1400
+ },
1401
+ "transformer_blocks.28.attn.to_out.0": {
1402
+ "in_features": 4096,
1403
+ "out_features": 4096,
1404
+ "rank": 128,
1405
+ "precision": "int4",
1406
+ "bias": false
1407
+ },
1408
+ "transformer_blocks.28.img_mlp.proj": {
1409
+ "in_features": 4096,
1410
+ "out_features": 12288,
1411
+ "rank": 128,
1412
+ "precision": "int4",
1413
+ "bias": false
1414
+ },
1415
+ "transformer_blocks.28.img_mlp.out": {
1416
+ "in_features": 12288,
1417
+ "out_features": 4096,
1418
+ "rank": 128,
1419
+ "precision": "int4",
1420
+ "bias": false
1421
+ },
1422
+ "transformer_blocks.28.img_mlp.gate_layer": {
1423
+ "in_features": 4096,
1424
+ "out_features": 12288,
1425
+ "rank": 128,
1426
+ "precision": "int4",
1427
+ "bias": false
1428
+ },
1429
+ "transformer_blocks.29.attn.to_q": {
1430
+ "in_features": 4096,
1431
+ "out_features": 4096,
1432
+ "rank": 128,
1433
+ "precision": "int4",
1434
+ "bias": false
1435
+ },
1436
+ "transformer_blocks.29.attn.to_k": {
1437
+ "in_features": 4096,
1438
+ "out_features": 4096,
1439
+ "rank": 128,
1440
+ "precision": "int4",
1441
+ "bias": false
1442
+ },
1443
+ "transformer_blocks.29.attn.to_v": {
1444
+ "in_features": 4096,
1445
+ "out_features": 4096,
1446
+ "rank": 128,
1447
+ "precision": "int4",
1448
+ "bias": false
1449
+ },
1450
+ "transformer_blocks.29.attn.to_out.0": {
1451
+ "in_features": 4096,
1452
+ "out_features": 4096,
1453
+ "rank": 128,
1454
+ "precision": "int4",
1455
+ "bias": false
1456
+ },
1457
+ "transformer_blocks.29.img_mlp.proj": {
1458
+ "in_features": 4096,
1459
+ "out_features": 12288,
1460
+ "rank": 128,
1461
+ "precision": "int4",
1462
+ "bias": false
1463
+ },
1464
+ "transformer_blocks.29.img_mlp.out": {
1465
+ "in_features": 12288,
1466
+ "out_features": 4096,
1467
+ "rank": 128,
1468
+ "precision": "int4",
1469
+ "bias": false
1470
+ },
1471
+ "transformer_blocks.29.img_mlp.gate_layer": {
1472
+ "in_features": 4096,
1473
+ "out_features": 12288,
1474
+ "rank": 128,
1475
+ "precision": "int4",
1476
+ "bias": false
1477
+ },
1478
+ "transformer_blocks.30.attn.to_q": {
1479
+ "in_features": 4096,
1480
+ "out_features": 4096,
1481
+ "rank": 128,
1482
+ "precision": "int4",
1483
+ "bias": false
1484
+ },
1485
+ "transformer_blocks.30.attn.to_k": {
1486
+ "in_features": 4096,
1487
+ "out_features": 4096,
1488
+ "rank": 128,
1489
+ "precision": "int4",
1490
+ "bias": false
1491
+ },
1492
+ "transformer_blocks.30.attn.to_v": {
1493
+ "in_features": 4096,
1494
+ "out_features": 4096,
1495
+ "rank": 128,
1496
+ "precision": "int4",
1497
+ "bias": false
1498
+ },
1499
+ "transformer_blocks.30.attn.to_out.0": {
1500
+ "in_features": 4096,
1501
+ "out_features": 4096,
1502
+ "rank": 128,
1503
+ "precision": "int4",
1504
+ "bias": false
1505
+ },
1506
+ "transformer_blocks.30.img_mlp.proj": {
1507
+ "in_features": 4096,
1508
+ "out_features": 12288,
1509
+ "rank": 128,
1510
+ "precision": "int4",
1511
+ "bias": false
1512
+ },
1513
+ "transformer_blocks.30.img_mlp.out": {
1514
+ "in_features": 12288,
1515
+ "out_features": 4096,
1516
+ "rank": 128,
1517
+ "precision": "int4",
1518
+ "bias": false
1519
+ },
1520
+ "transformer_blocks.30.img_mlp.gate_layer": {
1521
+ "in_features": 4096,
1522
+ "out_features": 12288,
1523
+ "rank": 128,
1524
+ "precision": "int4",
1525
+ "bias": false
1526
+ },
1527
+ "transformer_blocks.31.attn.to_q": {
1528
+ "in_features": 4096,
1529
+ "out_features": 4096,
1530
+ "rank": 128,
1531
+ "precision": "int4",
1532
+ "bias": false
1533
+ },
1534
+ "transformer_blocks.31.attn.to_k": {
1535
+ "in_features": 4096,
1536
+ "out_features": 4096,
1537
+ "rank": 128,
1538
+ "precision": "int4",
1539
+ "bias": false
1540
+ },
1541
+ "transformer_blocks.31.attn.to_v": {
1542
+ "in_features": 4096,
1543
+ "out_features": 4096,
1544
+ "rank": 128,
1545
+ "precision": "int4",
1546
+ "bias": false
1547
+ },
1548
+ "transformer_blocks.31.attn.to_out.0": {
1549
+ "in_features": 4096,
1550
+ "out_features": 4096,
1551
+ "rank": 128,
1552
+ "precision": "int4",
1553
+ "bias": false
1554
+ },
1555
+ "transformer_blocks.31.img_mlp.proj": {
1556
+ "in_features": 4096,
1557
+ "out_features": 12288,
1558
+ "rank": 128,
1559
+ "precision": "int4",
1560
+ "bias": false
1561
+ },
1562
+ "transformer_blocks.31.img_mlp.out": {
1563
+ "in_features": 12288,
1564
+ "out_features": 4096,
1565
+ "rank": 128,
1566
+ "precision": "int4",
1567
+ "bias": false
1568
+ },
1569
+ "transformer_blocks.31.img_mlp.gate_layer": {
1570
+ "in_features": 4096,
1571
+ "out_features": 12288,
1572
+ "rank": 128,
1573
+ "precision": "int4",
1574
+ "bias": false
1575
+ }
1576
+ },
1577
+ "layer_count": 224,
1578
+ "requires": {
1579
+ "torch": "2.12.1",
1580
+ "nunchaku": "1.3.0.dev20260306+cu13.0torch2.12"
1581
+ },
1582
+ "calibration": {
1583
+ "layers": 224,
1584
+ "alpha": 0.5,
1585
+ "rank": 128,
1586
+ "gptq": true,
1587
+ "method": "SmoothQuant smoothing + rank-r SVD + GPTQ-rounded 4-bit residual"
1588
+ },
1589
+ "report": {
1590
+ "sampled": 7,
1591
+ "beat_rtn": 0,
1592
+ "clipped": [],
1593
+ "layers": [
1594
+ {
1595
+ "layer": "transformer_blocks.0.attn.to_k",
1596
+ "method": "gptq",
1597
+ "precision": "int4",
1598
+ "dead_channels": 0,
1599
+ "max_drift": 1.0,
1600
+ "headroom": null,
1601
+ "clipped": false,
1602
+ "gptq_weight_error": 0.138597,
1603
+ "rtn_weight_error": 0.128178
1604
+ },
1605
+ {
1606
+ "layer": "transformer_blocks.12.img_mlp.gate_layer",
1607
+ "method": "gptq",
1608
+ "precision": "int4",
1609
+ "dead_channels": 0,
1610
+ "max_drift": 1.018,
1611
+ "headroom": null,
1612
+ "clipped": false,
1613
+ "gptq_weight_error": 0.152526,
1614
+ "rtn_weight_error": 0.1273
1615
+ },
1616
+ {
1617
+ "layer": "transformer_blocks.17.attn.to_out.0",
1618
+ "method": "gptq",
1619
+ "precision": "int4",
1620
+ "dead_channels": 0,
1621
+ "max_drift": 1.003,
1622
+ "headroom": null,
1623
+ "clipped": false,
1624
+ "gptq_weight_error": 0.141176,
1625
+ "rtn_weight_error": 0.117945
1626
+ },
1627
+ {
1628
+ "layer": "transformer_blocks.20.img_mlp.out",
1629
+ "method": "gptq",
1630
+ "precision": "int4",
1631
+ "dead_channels": 0,
1632
+ "max_drift": 1.016,
1633
+ "headroom": null,
1634
+ "clipped": false,
1635
+ "gptq_weight_error": 0.17955,
1636
+ "rtn_weight_error": 0.139057
1637
+ },
1638
+ {
1639
+ "layer": "transformer_blocks.25.attn.to_q",
1640
+ "method": "gptq",
1641
+ "precision": "int4",
1642
+ "dead_channels": 0,
1643
+ "max_drift": 1.013,
1644
+ "headroom": null,
1645
+ "clipped": false,
1646
+ "gptq_weight_error": 0.156766,
1647
+ "rtn_weight_error": 0.126401
1648
+ },
1649
+ {
1650
+ "layer": "transformer_blocks.29.img_mlp.proj",
1651
+ "method": "gptq",
1652
+ "precision": "int4",
1653
+ "dead_channels": 0,
1654
+ "max_drift": 1.019,
1655
+ "headroom": null,
1656
+ "clipped": false,
1657
+ "gptq_weight_error": 0.154812,
1658
+ "rtn_weight_error": 0.128077
1659
+ },
1660
+ {
1661
+ "layer": "transformer_blocks.5.attn.to_v",
1662
+ "method": "gptq",
1663
+ "precision": "int4",
1664
+ "dead_channels": 0,
1665
+ "max_drift": 1.007,
1666
+ "headroom": null,
1667
+ "clipped": false,
1668
+ "gptq_weight_error": 0.165468,
1669
+ "rtn_weight_error": 0.130482
1670
+ }
1671
+ ]
1672
+ }
1673
+ }
samples/flat.png ADDED

Git LFS Details

  • SHA256: 5f87042ca9f16479e05d8e4b59162e2bded35e8a192026edcdf233ee15474583
  • Pointer size: 132 Bytes
  • Size of remote file: 1.08 MB
samples/landscape.png ADDED

Git LFS Details

  • SHA256: febf174e10f343aa6b09ad24b14157622234af6bad689f48c667772c81ea9b95
  • Pointer size: 132 Bytes
  • Size of remote file: 1.94 MB
samples/orrery.png ADDED

Git LFS Details

  • SHA256: e8a1942fe7d0e2be5b02585557816f674fc1ac5e07f8c2aae662e308f3e5a4f5
  • Pointer size: 132 Bytes
  • Size of remote file: 1.71 MB
samples/portrait.png ADDED

Git LFS Details

  • SHA256: 132d53f8a3a3fa5d977f5875746e9109763f3b42cd632a77ae46c82ccdd17a2f
  • Pointer size: 132 Bytes
  • Size of remote file: 1.98 MB
samples/text-board.png ADDED

Git LFS Details

  • SHA256: b7e87e72c90991806f64aa39c8f1abd5625ad4d403353ccde8101281265037e2
  • Pointer size: 132 Bytes
  • Size of remote file: 1.96 MB
samples/text-sign.png ADDED

Git LFS Details

  • SHA256: b1a8200820c66db416e8ee20ba10706cf891a24ab8b26ef1ea950cc75ea63a3b
  • Pointer size: 132 Bytes
  • Size of remote file: 1.91 MB
w4a4_build.json ADDED
@@ -0,0 +1,161 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "Qwen/Qwen-Image-2.1",
3
+ "rank": 128,
4
+ "alpha": 0.5,
5
+ "gptq": true,
6
+ "scheme": "nvfp4",
7
+ "stages": {
8
+ "grid_check": {
9
+ "max_delta": 0.0,
10
+ "idempotent": true
11
+ },
12
+ "channel_stats": {
13
+ "layers": 224,
14
+ "seconds": 253.9
15
+ },
16
+ "input_groups": {
17
+ "groups": 128,
18
+ "shared_groups": 64,
19
+ "example": [
20
+ "transformer_blocks.0.attn.to_k",
21
+ "transformer_blocks.0.attn.to_q",
22
+ "transformer_blocks.0.attn.to_v"
23
+ ]
24
+ },
25
+ "hessians": {
26
+ "seconds": 300.6,
27
+ "token_sample": 512
28
+ },
29
+ "convert": {
30
+ "layers": 224,
31
+ "seconds": 17.3
32
+ },
33
+ "gptq": {
34
+ "sampled": 7,
35
+ "beat_rtn": 0,
36
+ "clipped": [],
37
+ "layers": [
38
+ {
39
+ "layer": "transformer_blocks.0.attn.to_k",
40
+ "method": "gptq",
41
+ "precision": "int4",
42
+ "dead_channels": 0,
43
+ "max_drift": 1.0,
44
+ "headroom": null,
45
+ "clipped": false,
46
+ "gptq_weight_error": 0.138597,
47
+ "rtn_weight_error": 0.128178
48
+ },
49
+ {
50
+ "layer": "transformer_blocks.12.img_mlp.gate_layer",
51
+ "method": "gptq",
52
+ "precision": "int4",
53
+ "dead_channels": 0,
54
+ "max_drift": 1.018,
55
+ "headroom": null,
56
+ "clipped": false,
57
+ "gptq_weight_error": 0.152526,
58
+ "rtn_weight_error": 0.1273
59
+ },
60
+ {
61
+ "layer": "transformer_blocks.17.attn.to_out.0",
62
+ "method": "gptq",
63
+ "precision": "int4",
64
+ "dead_channels": 0,
65
+ "max_drift": 1.003,
66
+ "headroom": null,
67
+ "clipped": false,
68
+ "gptq_weight_error": 0.141176,
69
+ "rtn_weight_error": 0.117945
70
+ },
71
+ {
72
+ "layer": "transformer_blocks.20.img_mlp.out",
73
+ "method": "gptq",
74
+ "precision": "int4",
75
+ "dead_channels": 0,
76
+ "max_drift": 1.016,
77
+ "headroom": null,
78
+ "clipped": false,
79
+ "gptq_weight_error": 0.17955,
80
+ "rtn_weight_error": 0.139057
81
+ },
82
+ {
83
+ "layer": "transformer_blocks.25.attn.to_q",
84
+ "method": "gptq",
85
+ "precision": "int4",
86
+ "dead_channels": 0,
87
+ "max_drift": 1.013,
88
+ "headroom": null,
89
+ "clipped": false,
90
+ "gptq_weight_error": 0.156766,
91
+ "rtn_weight_error": 0.126401
92
+ },
93
+ {
94
+ "layer": "transformer_blocks.29.img_mlp.proj",
95
+ "method": "gptq",
96
+ "precision": "int4",
97
+ "dead_channels": 0,
98
+ "max_drift": 1.019,
99
+ "headroom": null,
100
+ "clipped": false,
101
+ "gptq_weight_error": 0.154812,
102
+ "rtn_weight_error": 0.128077
103
+ },
104
+ {
105
+ "layer": "transformer_blocks.5.attn.to_v",
106
+ "method": "gptq",
107
+ "precision": "int4",
108
+ "dead_channels": 0,
109
+ "max_drift": 1.007,
110
+ "headroom": null,
111
+ "clipped": false,
112
+ "gptq_weight_error": 0.165468,
113
+ "rtn_weight_error": 0.130482
114
+ }
115
+ ]
116
+ },
117
+ "residual_quantize": {
118
+ "layers": 0,
119
+ "seconds": 0.0,
120
+ "resident_gb": 21.34
121
+ },
122
+ "save": {
123
+ "path": "/workspace/ckpt-int4",
124
+ "layers": 224,
125
+ "bytes": 4655346432,
126
+ "gb": 4.34
127
+ }
128
+ },
129
+ "calibration_set": {
130
+ "total": 30,
131
+ "text_to_image": 20,
132
+ "with_negative_prompt": 2,
133
+ "edits": 10,
134
+ "reference_images_total": 19,
135
+ "max_references": 5,
136
+ "sizes": [
137
+ [
138
+ 768,
139
+ 1344
140
+ ],
141
+ [
142
+ 832,
143
+ 1216
144
+ ],
145
+ [
146
+ 1024,
147
+ 1024
148
+ ],
149
+ [
150
+ 1216,
151
+ 832
152
+ ],
153
+ [
154
+ 1344,
155
+ 768
156
+ ]
157
+ ],
158
+ "estimated_seconds": 465
159
+ },
160
+ "reference_images": 6
161
+ }