vladmandic commited on
Commit
0d89766
·
verified ·
1 Parent(s): 20fab8e

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ 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
+ tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
model_index.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Pipeline",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "CalamitousFelicitousness/Krea-2-Base-Diffusers",
5
+ "scheduler": [
6
+ "diffusers",
7
+ "FlowMatchEulerDiscreteScheduler"
8
+ ],
9
+ "text_encoder": [
10
+ "transformers",
11
+ "Qwen3VLModel"
12
+ ],
13
+ "tokenizer": [
14
+ "transformers",
15
+ "Qwen2Tokenizer"
16
+ ],
17
+ "transformer": [
18
+ "pipelines.krea2.transformer_krea2",
19
+ "Krea2Transformer2DModel"
20
+ ],
21
+ "vae": [
22
+ "diffusers",
23
+ "AutoencoderKLQwenImage"
24
+ ]
25
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "base_image_seq_len": 256,
5
+ "base_shift": 0.5,
6
+ "invert_sigmas": false,
7
+ "max_image_seq_len": 6400,
8
+ "max_shift": 1.15,
9
+ "num_train_timesteps": 1000,
10
+ "shift": 1.0,
11
+ "shift_terminal": null,
12
+ "stochastic_sampling": false,
13
+ "time_shift_type": "exponential",
14
+ "use_beta_sigmas": false,
15
+ "use_dynamic_shifting": true,
16
+ "use_exponential_sigmas": false,
17
+ "use_karras_sigmas": false
18
+ }
text_encoder/config.json ADDED
@@ -0,0 +1,682 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3VLModel"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "image_token_id": 151655,
7
+ "model_type": "qwen3_vl",
8
+ "quantization_config": {
9
+ "add_skip_keys": true,
10
+ "dequantize_fp32": false,
11
+ "dynamic_loss_threshold": -1,
12
+ "group_size": 0,
13
+ "hadamard_group_size": 256,
14
+ "is_integer": true,
15
+ "is_training": false,
16
+ "modules_dtype_dict": {
17
+ "uint4": [
18
+ "language_model.layers.0.mlp.down_proj.weight",
19
+ "language_model.layers.0.mlp.gate_proj.weight",
20
+ "language_model.layers.0.mlp.up_proj.weight",
21
+ "language_model.layers.0.self_attn.k_proj.weight",
22
+ "language_model.layers.0.self_attn.o_proj.weight",
23
+ "language_model.layers.0.self_attn.q_proj.weight",
24
+ "language_model.layers.0.self_attn.v_proj.weight",
25
+ "language_model.layers.1.mlp.down_proj.weight",
26
+ "language_model.layers.1.mlp.gate_proj.weight",
27
+ "language_model.layers.1.mlp.up_proj.weight",
28
+ "language_model.layers.1.self_attn.k_proj.weight",
29
+ "language_model.layers.1.self_attn.o_proj.weight",
30
+ "language_model.layers.1.self_attn.q_proj.weight",
31
+ "language_model.layers.1.self_attn.v_proj.weight",
32
+ "language_model.layers.2.mlp.down_proj.weight",
33
+ "language_model.layers.2.mlp.gate_proj.weight",
34
+ "language_model.layers.2.mlp.up_proj.weight",
35
+ "language_model.layers.2.self_attn.k_proj.weight",
36
+ "language_model.layers.2.self_attn.o_proj.weight",
37
+ "language_model.layers.2.self_attn.q_proj.weight",
38
+ "language_model.layers.2.self_attn.v_proj.weight",
39
+ "language_model.layers.3.mlp.down_proj.weight",
40
+ "language_model.layers.3.mlp.gate_proj.weight",
41
+ "language_model.layers.3.mlp.up_proj.weight",
42
+ "language_model.layers.3.self_attn.k_proj.weight",
43
+ "language_model.layers.3.self_attn.o_proj.weight",
44
+ "language_model.layers.3.self_attn.q_proj.weight",
45
+ "language_model.layers.3.self_attn.v_proj.weight",
46
+ "language_model.layers.4.mlp.down_proj.weight",
47
+ "language_model.layers.4.mlp.gate_proj.weight",
48
+ "language_model.layers.4.mlp.up_proj.weight",
49
+ "language_model.layers.4.self_attn.k_proj.weight",
50
+ "language_model.layers.4.self_attn.o_proj.weight",
51
+ "language_model.layers.4.self_attn.q_proj.weight",
52
+ "language_model.layers.4.self_attn.v_proj.weight",
53
+ "language_model.layers.5.mlp.down_proj.weight",
54
+ "language_model.layers.5.mlp.gate_proj.weight",
55
+ "language_model.layers.5.mlp.up_proj.weight",
56
+ "language_model.layers.5.self_attn.k_proj.weight",
57
+ "language_model.layers.5.self_attn.o_proj.weight",
58
+ "language_model.layers.5.self_attn.q_proj.weight",
59
+ "language_model.layers.5.self_attn.v_proj.weight",
60
+ "language_model.layers.6.mlp.down_proj.weight",
61
+ "language_model.layers.6.mlp.gate_proj.weight",
62
+ "language_model.layers.6.mlp.up_proj.weight",
63
+ "language_model.layers.6.self_attn.k_proj.weight",
64
+ "language_model.layers.6.self_attn.o_proj.weight",
65
+ "language_model.layers.6.self_attn.q_proj.weight",
66
+ "language_model.layers.6.self_attn.v_proj.weight",
67
+ "language_model.layers.7.mlp.down_proj.weight",
68
+ "language_model.layers.7.mlp.gate_proj.weight",
69
+ "language_model.layers.7.mlp.up_proj.weight",
70
+ "language_model.layers.7.self_attn.k_proj.weight",
71
+ "language_model.layers.7.self_attn.o_proj.weight",
72
+ "language_model.layers.7.self_attn.q_proj.weight",
73
+ "language_model.layers.7.self_attn.v_proj.weight",
74
+ "language_model.layers.8.mlp.down_proj.weight",
75
+ "language_model.layers.8.mlp.gate_proj.weight",
76
+ "language_model.layers.8.mlp.up_proj.weight",
77
+ "language_model.layers.8.self_attn.k_proj.weight",
78
+ "language_model.layers.8.self_attn.o_proj.weight",
79
+ "language_model.layers.8.self_attn.q_proj.weight",
80
+ "language_model.layers.8.self_attn.v_proj.weight",
81
+ "language_model.layers.9.mlp.down_proj.weight",
82
+ "language_model.layers.9.mlp.gate_proj.weight",
83
+ "language_model.layers.9.mlp.up_proj.weight",
84
+ "language_model.layers.9.self_attn.k_proj.weight",
85
+ "language_model.layers.9.self_attn.o_proj.weight",
86
+ "language_model.layers.9.self_attn.q_proj.weight",
87
+ "language_model.layers.9.self_attn.v_proj.weight",
88
+ "language_model.layers.10.mlp.down_proj.weight",
89
+ "language_model.layers.10.mlp.gate_proj.weight",
90
+ "language_model.layers.10.mlp.up_proj.weight",
91
+ "language_model.layers.10.self_attn.k_proj.weight",
92
+ "language_model.layers.10.self_attn.o_proj.weight",
93
+ "language_model.layers.10.self_attn.q_proj.weight",
94
+ "language_model.layers.10.self_attn.v_proj.weight",
95
+ "language_model.layers.11.mlp.down_proj.weight",
96
+ "language_model.layers.11.mlp.gate_proj.weight",
97
+ "language_model.layers.11.mlp.up_proj.weight",
98
+ "language_model.layers.11.self_attn.k_proj.weight",
99
+ "language_model.layers.11.self_attn.o_proj.weight",
100
+ "language_model.layers.11.self_attn.q_proj.weight",
101
+ "language_model.layers.11.self_attn.v_proj.weight",
102
+ "language_model.layers.12.mlp.down_proj.weight",
103
+ "language_model.layers.12.mlp.gate_proj.weight",
104
+ "language_model.layers.12.mlp.up_proj.weight",
105
+ "language_model.layers.12.self_attn.k_proj.weight",
106
+ "language_model.layers.12.self_attn.o_proj.weight",
107
+ "language_model.layers.12.self_attn.q_proj.weight",
108
+ "language_model.layers.12.self_attn.v_proj.weight",
109
+ "language_model.layers.13.mlp.down_proj.weight",
110
+ "language_model.layers.13.mlp.gate_proj.weight",
111
+ "language_model.layers.13.mlp.up_proj.weight",
112
+ "language_model.layers.13.self_attn.k_proj.weight",
113
+ "language_model.layers.13.self_attn.o_proj.weight",
114
+ "language_model.layers.13.self_attn.q_proj.weight",
115
+ "language_model.layers.13.self_attn.v_proj.weight",
116
+ "language_model.layers.14.mlp.down_proj.weight",
117
+ "language_model.layers.14.mlp.gate_proj.weight",
118
+ "language_model.layers.14.mlp.up_proj.weight",
119
+ "language_model.layers.14.self_attn.k_proj.weight",
120
+ "language_model.layers.14.self_attn.o_proj.weight",
121
+ "language_model.layers.14.self_attn.q_proj.weight",
122
+ "language_model.layers.14.self_attn.v_proj.weight",
123
+ "language_model.layers.15.mlp.down_proj.weight",
124
+ "language_model.layers.15.mlp.gate_proj.weight",
125
+ "language_model.layers.15.mlp.up_proj.weight",
126
+ "language_model.layers.15.self_attn.k_proj.weight",
127
+ "language_model.layers.15.self_attn.o_proj.weight",
128
+ "language_model.layers.15.self_attn.q_proj.weight",
129
+ "language_model.layers.15.self_attn.v_proj.weight",
130
+ "language_model.layers.16.mlp.down_proj.weight",
131
+ "language_model.layers.16.mlp.gate_proj.weight",
132
+ "language_model.layers.16.mlp.up_proj.weight",
133
+ "language_model.layers.16.self_attn.k_proj.weight",
134
+ "language_model.layers.16.self_attn.o_proj.weight",
135
+ "language_model.layers.16.self_attn.q_proj.weight",
136
+ "language_model.layers.16.self_attn.v_proj.weight",
137
+ "language_model.layers.17.mlp.down_proj.weight",
138
+ "language_model.layers.17.mlp.gate_proj.weight",
139
+ "language_model.layers.17.mlp.up_proj.weight",
140
+ "language_model.layers.17.self_attn.k_proj.weight",
141
+ "language_model.layers.17.self_attn.o_proj.weight",
142
+ "language_model.layers.17.self_attn.q_proj.weight",
143
+ "language_model.layers.17.self_attn.v_proj.weight",
144
+ "language_model.layers.18.mlp.down_proj.weight",
145
+ "language_model.layers.18.mlp.gate_proj.weight",
146
+ "language_model.layers.18.mlp.up_proj.weight",
147
+ "language_model.layers.18.self_attn.k_proj.weight",
148
+ "language_model.layers.18.self_attn.o_proj.weight",
149
+ "language_model.layers.18.self_attn.q_proj.weight",
150
+ "language_model.layers.18.self_attn.v_proj.weight",
151
+ "language_model.layers.19.mlp.down_proj.weight",
152
+ "language_model.layers.19.mlp.gate_proj.weight",
153
+ "language_model.layers.19.mlp.up_proj.weight",
154
+ "language_model.layers.19.self_attn.k_proj.weight",
155
+ "language_model.layers.19.self_attn.o_proj.weight",
156
+ "language_model.layers.19.self_attn.q_proj.weight",
157
+ "language_model.layers.19.self_attn.v_proj.weight",
158
+ "language_model.layers.20.mlp.down_proj.weight",
159
+ "language_model.layers.20.mlp.gate_proj.weight",
160
+ "language_model.layers.20.mlp.up_proj.weight",
161
+ "language_model.layers.20.self_attn.k_proj.weight",
162
+ "language_model.layers.20.self_attn.o_proj.weight",
163
+ "language_model.layers.20.self_attn.q_proj.weight",
164
+ "language_model.layers.20.self_attn.v_proj.weight",
165
+ "language_model.layers.21.mlp.down_proj.weight",
166
+ "language_model.layers.21.mlp.gate_proj.weight",
167
+ "language_model.layers.21.mlp.up_proj.weight",
168
+ "language_model.layers.21.self_attn.k_proj.weight",
169
+ "language_model.layers.21.self_attn.o_proj.weight",
170
+ "language_model.layers.21.self_attn.q_proj.weight",
171
+ "language_model.layers.21.self_attn.v_proj.weight",
172
+ "language_model.layers.22.mlp.down_proj.weight",
173
+ "language_model.layers.22.mlp.gate_proj.weight",
174
+ "language_model.layers.22.mlp.up_proj.weight",
175
+ "language_model.layers.22.self_attn.k_proj.weight",
176
+ "language_model.layers.22.self_attn.o_proj.weight",
177
+ "language_model.layers.22.self_attn.q_proj.weight",
178
+ "language_model.layers.22.self_attn.v_proj.weight",
179
+ "language_model.layers.23.mlp.down_proj.weight",
180
+ "language_model.layers.23.mlp.gate_proj.weight",
181
+ "language_model.layers.23.mlp.up_proj.weight",
182
+ "language_model.layers.23.self_attn.k_proj.weight",
183
+ "language_model.layers.23.self_attn.o_proj.weight",
184
+ "language_model.layers.23.self_attn.q_proj.weight",
185
+ "language_model.layers.23.self_attn.v_proj.weight",
186
+ "language_model.layers.24.mlp.down_proj.weight",
187
+ "language_model.layers.24.mlp.gate_proj.weight",
188
+ "language_model.layers.24.mlp.up_proj.weight",
189
+ "language_model.layers.24.self_attn.k_proj.weight",
190
+ "language_model.layers.24.self_attn.o_proj.weight",
191
+ "language_model.layers.24.self_attn.q_proj.weight",
192
+ "language_model.layers.24.self_attn.v_proj.weight",
193
+ "language_model.layers.25.mlp.down_proj.weight",
194
+ "language_model.layers.25.mlp.gate_proj.weight",
195
+ "language_model.layers.25.mlp.up_proj.weight",
196
+ "language_model.layers.25.self_attn.k_proj.weight",
197
+ "language_model.layers.25.self_attn.o_proj.weight",
198
+ "language_model.layers.25.self_attn.q_proj.weight",
199
+ "language_model.layers.25.self_attn.v_proj.weight",
200
+ "language_model.layers.26.mlp.down_proj.weight",
201
+ "language_model.layers.26.mlp.gate_proj.weight",
202
+ "language_model.layers.26.mlp.up_proj.weight",
203
+ "language_model.layers.26.self_attn.k_proj.weight",
204
+ "language_model.layers.26.self_attn.o_proj.weight",
205
+ "language_model.layers.26.self_attn.q_proj.weight",
206
+ "language_model.layers.26.self_attn.v_proj.weight",
207
+ "language_model.layers.27.mlp.down_proj.weight",
208
+ "language_model.layers.27.mlp.gate_proj.weight",
209
+ "language_model.layers.27.mlp.up_proj.weight",
210
+ "language_model.layers.27.self_attn.k_proj.weight",
211
+ "language_model.layers.27.self_attn.o_proj.weight",
212
+ "language_model.layers.27.self_attn.q_proj.weight",
213
+ "language_model.layers.27.self_attn.v_proj.weight",
214
+ "language_model.layers.28.mlp.down_proj.weight",
215
+ "language_model.layers.28.mlp.gate_proj.weight",
216
+ "language_model.layers.28.mlp.up_proj.weight",
217
+ "language_model.layers.28.self_attn.k_proj.weight",
218
+ "language_model.layers.28.self_attn.o_proj.weight",
219
+ "language_model.layers.28.self_attn.q_proj.weight",
220
+ "language_model.layers.28.self_attn.v_proj.weight",
221
+ "language_model.layers.29.mlp.down_proj.weight",
222
+ "language_model.layers.29.mlp.gate_proj.weight",
223
+ "language_model.layers.29.mlp.up_proj.weight",
224
+ "language_model.layers.29.self_attn.k_proj.weight",
225
+ "language_model.layers.29.self_attn.o_proj.weight",
226
+ "language_model.layers.29.self_attn.q_proj.weight",
227
+ "language_model.layers.29.self_attn.v_proj.weight",
228
+ "language_model.layers.30.mlp.down_proj.weight",
229
+ "language_model.layers.30.mlp.gate_proj.weight",
230
+ "language_model.layers.30.mlp.up_proj.weight",
231
+ "language_model.layers.30.self_attn.k_proj.weight",
232
+ "language_model.layers.30.self_attn.o_proj.weight",
233
+ "language_model.layers.30.self_attn.q_proj.weight",
234
+ "language_model.layers.30.self_attn.v_proj.weight",
235
+ "language_model.layers.31.mlp.down_proj.weight",
236
+ "language_model.layers.31.mlp.gate_proj.weight",
237
+ "language_model.layers.31.mlp.up_proj.weight",
238
+ "language_model.layers.31.self_attn.k_proj.weight",
239
+ "language_model.layers.31.self_attn.o_proj.weight",
240
+ "language_model.layers.31.self_attn.q_proj.weight",
241
+ "language_model.layers.31.self_attn.v_proj.weight",
242
+ "language_model.layers.32.mlp.down_proj.weight",
243
+ "language_model.layers.32.mlp.gate_proj.weight",
244
+ "language_model.layers.32.mlp.up_proj.weight",
245
+ "language_model.layers.32.self_attn.k_proj.weight",
246
+ "language_model.layers.32.self_attn.o_proj.weight",
247
+ "language_model.layers.32.self_attn.q_proj.weight",
248
+ "language_model.layers.32.self_attn.v_proj.weight",
249
+ "language_model.layers.33.mlp.down_proj.weight",
250
+ "language_model.layers.33.mlp.gate_proj.weight",
251
+ "language_model.layers.33.mlp.up_proj.weight",
252
+ "language_model.layers.33.self_attn.k_proj.weight",
253
+ "language_model.layers.33.self_attn.o_proj.weight",
254
+ "language_model.layers.33.self_attn.q_proj.weight",
255
+ "language_model.layers.33.self_attn.v_proj.weight",
256
+ "language_model.layers.34.mlp.down_proj.weight",
257
+ "language_model.layers.34.mlp.gate_proj.weight",
258
+ "language_model.layers.34.mlp.up_proj.weight",
259
+ "language_model.layers.34.self_attn.k_proj.weight",
260
+ "language_model.layers.34.self_attn.o_proj.weight",
261
+ "language_model.layers.34.self_attn.q_proj.weight",
262
+ "language_model.layers.34.self_attn.v_proj.weight",
263
+ "language_model.layers.35.mlp.down_proj.weight",
264
+ "language_model.layers.35.mlp.gate_proj.weight",
265
+ "language_model.layers.35.mlp.up_proj.weight",
266
+ "language_model.layers.35.self_attn.k_proj.weight",
267
+ "language_model.layers.35.self_attn.o_proj.weight",
268
+ "language_model.layers.35.self_attn.q_proj.weight",
269
+ "language_model.layers.35.self_attn.v_proj.weight",
270
+ "visual.blocks.0.attn.proj.weight",
271
+ "visual.blocks.0.attn.qkv.weight",
272
+ "visual.blocks.0.mlp.linear_fc1.weight",
273
+ "visual.blocks.0.mlp.linear_fc2.weight",
274
+ "visual.blocks.1.attn.proj.weight",
275
+ "visual.blocks.1.attn.qkv.weight",
276
+ "visual.blocks.1.mlp.linear_fc1.weight",
277
+ "visual.blocks.1.mlp.linear_fc2.weight",
278
+ "visual.blocks.2.attn.proj.weight",
279
+ "visual.blocks.2.attn.qkv.weight",
280
+ "visual.blocks.2.mlp.linear_fc1.weight",
281
+ "visual.blocks.2.mlp.linear_fc2.weight",
282
+ "visual.blocks.3.attn.proj.weight",
283
+ "visual.blocks.3.attn.qkv.weight",
284
+ "visual.blocks.3.mlp.linear_fc1.weight",
285
+ "visual.blocks.3.mlp.linear_fc2.weight",
286
+ "visual.blocks.4.attn.proj.weight",
287
+ "visual.blocks.4.attn.qkv.weight",
288
+ "visual.blocks.4.mlp.linear_fc1.weight",
289
+ "visual.blocks.4.mlp.linear_fc2.weight",
290
+ "visual.blocks.5.attn.proj.weight",
291
+ "visual.blocks.5.attn.qkv.weight",
292
+ "visual.blocks.5.mlp.linear_fc1.weight",
293
+ "visual.blocks.5.mlp.linear_fc2.weight",
294
+ "visual.blocks.6.attn.proj.weight",
295
+ "visual.blocks.6.attn.qkv.weight",
296
+ "visual.blocks.6.mlp.linear_fc1.weight",
297
+ "visual.blocks.6.mlp.linear_fc2.weight",
298
+ "visual.blocks.7.attn.proj.weight",
299
+ "visual.blocks.7.attn.qkv.weight",
300
+ "visual.blocks.7.mlp.linear_fc1.weight",
301
+ "visual.blocks.7.mlp.linear_fc2.weight",
302
+ "visual.blocks.8.attn.proj.weight",
303
+ "visual.blocks.8.attn.qkv.weight",
304
+ "visual.blocks.8.mlp.linear_fc1.weight",
305
+ "visual.blocks.8.mlp.linear_fc2.weight",
306
+ "visual.blocks.9.attn.proj.weight",
307
+ "visual.blocks.9.attn.qkv.weight",
308
+ "visual.blocks.9.mlp.linear_fc1.weight",
309
+ "visual.blocks.9.mlp.linear_fc2.weight",
310
+ "visual.blocks.10.attn.proj.weight",
311
+ "visual.blocks.10.attn.qkv.weight",
312
+ "visual.blocks.10.mlp.linear_fc1.weight",
313
+ "visual.blocks.10.mlp.linear_fc2.weight",
314
+ "visual.blocks.11.attn.proj.weight",
315
+ "visual.blocks.11.attn.qkv.weight",
316
+ "visual.blocks.11.mlp.linear_fc1.weight",
317
+ "visual.blocks.11.mlp.linear_fc2.weight",
318
+ "visual.blocks.12.attn.proj.weight",
319
+ "visual.blocks.12.attn.qkv.weight",
320
+ "visual.blocks.12.mlp.linear_fc1.weight",
321
+ "visual.blocks.12.mlp.linear_fc2.weight",
322
+ "visual.blocks.13.attn.proj.weight",
323
+ "visual.blocks.13.attn.qkv.weight",
324
+ "visual.blocks.13.mlp.linear_fc1.weight",
325
+ "visual.blocks.13.mlp.linear_fc2.weight",
326
+ "visual.blocks.14.attn.proj.weight",
327
+ "visual.blocks.14.attn.qkv.weight",
328
+ "visual.blocks.14.mlp.linear_fc1.weight",
329
+ "visual.blocks.14.mlp.linear_fc2.weight",
330
+ "visual.blocks.15.attn.proj.weight",
331
+ "visual.blocks.15.attn.qkv.weight",
332
+ "visual.blocks.15.mlp.linear_fc1.weight",
333
+ "visual.blocks.15.mlp.linear_fc2.weight",
334
+ "visual.blocks.16.attn.proj.weight",
335
+ "visual.blocks.16.attn.qkv.weight",
336
+ "visual.blocks.16.mlp.linear_fc1.weight",
337
+ "visual.blocks.16.mlp.linear_fc2.weight",
338
+ "visual.blocks.17.attn.proj.weight",
339
+ "visual.blocks.17.attn.qkv.weight",
340
+ "visual.blocks.17.mlp.linear_fc1.weight",
341
+ "visual.blocks.17.mlp.linear_fc2.weight",
342
+ "visual.blocks.18.attn.proj.weight",
343
+ "visual.blocks.18.attn.qkv.weight",
344
+ "visual.blocks.18.mlp.linear_fc1.weight",
345
+ "visual.blocks.18.mlp.linear_fc2.weight",
346
+ "visual.blocks.19.attn.proj.weight",
347
+ "visual.blocks.19.attn.qkv.weight",
348
+ "visual.blocks.19.mlp.linear_fc1.weight",
349
+ "visual.blocks.19.mlp.linear_fc2.weight",
350
+ "visual.blocks.20.attn.proj.weight",
351
+ "visual.blocks.20.attn.qkv.weight",
352
+ "visual.blocks.20.mlp.linear_fc1.weight",
353
+ "visual.blocks.20.mlp.linear_fc2.weight",
354
+ "visual.blocks.21.attn.proj.weight",
355
+ "visual.blocks.21.attn.qkv.weight",
356
+ "visual.blocks.21.mlp.linear_fc1.weight",
357
+ "visual.blocks.21.mlp.linear_fc2.weight",
358
+ "visual.blocks.22.attn.proj.weight",
359
+ "visual.blocks.22.attn.qkv.weight",
360
+ "visual.blocks.22.mlp.linear_fc1.weight",
361
+ "visual.blocks.22.mlp.linear_fc2.weight",
362
+ "visual.blocks.23.attn.proj.weight",
363
+ "visual.blocks.23.attn.qkv.weight",
364
+ "visual.blocks.23.mlp.linear_fc1.weight",
365
+ "visual.blocks.23.mlp.linear_fc2.weight",
366
+ "visual.deepstack_merger_list.0.linear_fc1.weight",
367
+ "visual.deepstack_merger_list.0.linear_fc2.weight",
368
+ "visual.deepstack_merger_list.1.linear_fc1.weight",
369
+ "visual.deepstack_merger_list.1.linear_fc2.weight",
370
+ "visual.deepstack_merger_list.2.linear_fc1.weight",
371
+ "visual.deepstack_merger_list.2.linear_fc2.weight",
372
+ "visual.merger.linear_fc1.weight",
373
+ "visual.merger.linear_fc2.weight"
374
+ ]
375
+ },
376
+ "modules_quant_config": {},
377
+ "modules_to_not_convert": [
378
+ ".txt_out",
379
+ ".time_embed",
380
+ ".vid_out",
381
+ ".vid_in",
382
+ ".txt_in",
383
+ ".img_in",
384
+ ".emb_out",
385
+ ".norm_out",
386
+ "multi_modal_projector",
387
+ ".img_out",
388
+ ".t_embedder",
389
+ ".final_layer",
390
+ ".emb_in",
391
+ "patch_embedding",
392
+ "language_model.norm",
393
+ ".proj_out",
394
+ "time_text_embed",
395
+ "patch_embed",
396
+ ".x_embedder",
397
+ "patch_emb",
398
+ "wte",
399
+ ".y_embedder",
400
+ ".condition_embedder",
401
+ ".context_embedder",
402
+ "lm_head",
403
+ "language_model.embed_tokens.weight",
404
+ "language_model.layers.0.input_layernorm.weight",
405
+ "language_model.layers.0.post_attention_layernorm.weight",
406
+ "language_model.layers.0.self_attn.k_norm.weight",
407
+ "language_model.layers.0.self_attn.q_norm.weight",
408
+ "language_model.layers.1.input_layernorm.weight",
409
+ "language_model.layers.1.post_attention_layernorm.weight",
410
+ "language_model.layers.1.self_attn.k_norm.weight",
411
+ "language_model.layers.1.self_attn.q_norm.weight",
412
+ "language_model.layers.2.input_layernorm.weight",
413
+ "language_model.layers.2.post_attention_layernorm.weight",
414
+ "language_model.layers.2.self_attn.k_norm.weight",
415
+ "language_model.layers.2.self_attn.q_norm.weight",
416
+ "language_model.layers.3.input_layernorm.weight",
417
+ "language_model.layers.3.post_attention_layernorm.weight",
418
+ "language_model.layers.3.self_attn.k_norm.weight",
419
+ "language_model.layers.3.self_attn.q_norm.weight",
420
+ "language_model.layers.4.input_layernorm.weight",
421
+ "language_model.layers.4.post_attention_layernorm.weight",
422
+ "language_model.layers.4.self_attn.k_norm.weight",
423
+ "language_model.layers.4.self_attn.q_norm.weight",
424
+ "language_model.layers.5.input_layernorm.weight",
425
+ "language_model.layers.5.post_attention_layernorm.weight",
426
+ "language_model.layers.5.self_attn.k_norm.weight",
427
+ "language_model.layers.5.self_attn.q_norm.weight",
428
+ "language_model.layers.6.input_layernorm.weight",
429
+ "language_model.layers.6.post_attention_layernorm.weight",
430
+ "language_model.layers.6.self_attn.k_norm.weight",
431
+ "language_model.layers.6.self_attn.q_norm.weight",
432
+ "language_model.layers.7.input_layernorm.weight",
433
+ "language_model.layers.7.post_attention_layernorm.weight",
434
+ "language_model.layers.7.self_attn.k_norm.weight",
435
+ "language_model.layers.7.self_attn.q_norm.weight",
436
+ "language_model.layers.8.input_layernorm.weight",
437
+ "language_model.layers.8.post_attention_layernorm.weight",
438
+ "language_model.layers.8.self_attn.k_norm.weight",
439
+ "language_model.layers.8.self_attn.q_norm.weight",
440
+ "language_model.layers.9.input_layernorm.weight",
441
+ "language_model.layers.9.post_attention_layernorm.weight",
442
+ "language_model.layers.9.self_attn.k_norm.weight",
443
+ "language_model.layers.9.self_attn.q_norm.weight",
444
+ "language_model.layers.10.input_layernorm.weight",
445
+ "language_model.layers.10.post_attention_layernorm.weight",
446
+ "language_model.layers.10.self_attn.k_norm.weight",
447
+ "language_model.layers.10.self_attn.q_norm.weight",
448
+ "language_model.layers.11.input_layernorm.weight",
449
+ "language_model.layers.11.post_attention_layernorm.weight",
450
+ "language_model.layers.11.self_attn.k_norm.weight",
451
+ "language_model.layers.11.self_attn.q_norm.weight",
452
+ "language_model.layers.12.input_layernorm.weight",
453
+ "language_model.layers.12.post_attention_layernorm.weight",
454
+ "language_model.layers.12.self_attn.k_norm.weight",
455
+ "language_model.layers.12.self_attn.q_norm.weight",
456
+ "language_model.layers.13.input_layernorm.weight",
457
+ "language_model.layers.13.post_attention_layernorm.weight",
458
+ "language_model.layers.13.self_attn.k_norm.weight",
459
+ "language_model.layers.13.self_attn.q_norm.weight",
460
+ "language_model.layers.14.input_layernorm.weight",
461
+ "language_model.layers.14.post_attention_layernorm.weight",
462
+ "language_model.layers.14.self_attn.k_norm.weight",
463
+ "language_model.layers.14.self_attn.q_norm.weight",
464
+ "language_model.layers.15.input_layernorm.weight",
465
+ "language_model.layers.15.post_attention_layernorm.weight",
466
+ "language_model.layers.15.self_attn.k_norm.weight",
467
+ "language_model.layers.15.self_attn.q_norm.weight",
468
+ "language_model.layers.16.input_layernorm.weight",
469
+ "language_model.layers.16.post_attention_layernorm.weight",
470
+ "language_model.layers.16.self_attn.k_norm.weight",
471
+ "language_model.layers.16.self_attn.q_norm.weight",
472
+ "language_model.layers.17.input_layernorm.weight",
473
+ "language_model.layers.17.post_attention_layernorm.weight",
474
+ "language_model.layers.17.self_attn.k_norm.weight",
475
+ "language_model.layers.17.self_attn.q_norm.weight",
476
+ "language_model.layers.18.input_layernorm.weight",
477
+ "language_model.layers.18.post_attention_layernorm.weight",
478
+ "language_model.layers.18.self_attn.k_norm.weight",
479
+ "language_model.layers.18.self_attn.q_norm.weight",
480
+ "language_model.layers.19.input_layernorm.weight",
481
+ "language_model.layers.19.post_attention_layernorm.weight",
482
+ "language_model.layers.19.self_attn.k_norm.weight",
483
+ "language_model.layers.19.self_attn.q_norm.weight",
484
+ "language_model.layers.20.input_layernorm.weight",
485
+ "language_model.layers.20.post_attention_layernorm.weight",
486
+ "language_model.layers.20.self_attn.k_norm.weight",
487
+ "language_model.layers.20.self_attn.q_norm.weight",
488
+ "language_model.layers.21.input_layernorm.weight",
489
+ "language_model.layers.21.post_attention_layernorm.weight",
490
+ "language_model.layers.21.self_attn.k_norm.weight",
491
+ "language_model.layers.21.self_attn.q_norm.weight",
492
+ "language_model.layers.22.input_layernorm.weight",
493
+ "language_model.layers.22.post_attention_layernorm.weight",
494
+ "language_model.layers.22.self_attn.k_norm.weight",
495
+ "language_model.layers.22.self_attn.q_norm.weight",
496
+ "language_model.layers.23.input_layernorm.weight",
497
+ "language_model.layers.23.post_attention_layernorm.weight",
498
+ "language_model.layers.23.self_attn.k_norm.weight",
499
+ "language_model.layers.23.self_attn.q_norm.weight",
500
+ "language_model.layers.24.input_layernorm.weight",
501
+ "language_model.layers.24.post_attention_layernorm.weight",
502
+ "language_model.layers.24.self_attn.k_norm.weight",
503
+ "language_model.layers.24.self_attn.q_norm.weight",
504
+ "language_model.layers.25.input_layernorm.weight",
505
+ "language_model.layers.25.post_attention_layernorm.weight",
506
+ "language_model.layers.25.self_attn.k_norm.weight",
507
+ "language_model.layers.25.self_attn.q_norm.weight",
508
+ "language_model.layers.26.input_layernorm.weight",
509
+ "language_model.layers.26.post_attention_layernorm.weight",
510
+ "language_model.layers.26.self_attn.k_norm.weight",
511
+ "language_model.layers.26.self_attn.q_norm.weight",
512
+ "language_model.layers.27.input_layernorm.weight",
513
+ "language_model.layers.27.post_attention_layernorm.weight",
514
+ "language_model.layers.27.self_attn.k_norm.weight",
515
+ "language_model.layers.27.self_attn.q_norm.weight",
516
+ "language_model.layers.28.input_layernorm.weight",
517
+ "language_model.layers.28.post_attention_layernorm.weight",
518
+ "language_model.layers.28.self_attn.k_norm.weight",
519
+ "language_model.layers.28.self_attn.q_norm.weight",
520
+ "language_model.layers.29.input_layernorm.weight",
521
+ "language_model.layers.29.post_attention_layernorm.weight",
522
+ "language_model.layers.29.self_attn.k_norm.weight",
523
+ "language_model.layers.29.self_attn.q_norm.weight",
524
+ "language_model.layers.30.input_layernorm.weight",
525
+ "language_model.layers.30.post_attention_layernorm.weight",
526
+ "language_model.layers.30.self_attn.k_norm.weight",
527
+ "language_model.layers.30.self_attn.q_norm.weight",
528
+ "language_model.layers.31.input_layernorm.weight",
529
+ "language_model.layers.31.post_attention_layernorm.weight",
530
+ "language_model.layers.31.self_attn.k_norm.weight",
531
+ "language_model.layers.31.self_attn.q_norm.weight",
532
+ "language_model.layers.32.input_layernorm.weight",
533
+ "language_model.layers.32.post_attention_layernorm.weight",
534
+ "language_model.layers.32.self_attn.k_norm.weight",
535
+ "language_model.layers.32.self_attn.q_norm.weight",
536
+ "language_model.layers.33.input_layernorm.weight",
537
+ "language_model.layers.33.post_attention_layernorm.weight",
538
+ "language_model.layers.33.self_attn.k_norm.weight",
539
+ "language_model.layers.33.self_attn.q_norm.weight",
540
+ "language_model.layers.34.input_layernorm.weight",
541
+ "language_model.layers.34.post_attention_layernorm.weight",
542
+ "language_model.layers.34.self_attn.k_norm.weight",
543
+ "language_model.layers.34.self_attn.q_norm.weight",
544
+ "language_model.layers.35.input_layernorm.weight",
545
+ "language_model.layers.35.post_attention_layernorm.weight",
546
+ "language_model.layers.35.self_attn.k_norm.weight",
547
+ "language_model.layers.35.self_attn.q_norm.weight",
548
+ "language_model.norm.weight",
549
+ "visual.blocks.0.norm1.weight",
550
+ "visual.blocks.0.norm2.weight",
551
+ "visual.blocks.1.norm1.weight",
552
+ "visual.blocks.1.norm2.weight",
553
+ "visual.blocks.2.norm1.weight",
554
+ "visual.blocks.2.norm2.weight",
555
+ "visual.blocks.3.norm1.weight",
556
+ "visual.blocks.3.norm2.weight",
557
+ "visual.blocks.4.norm1.weight",
558
+ "visual.blocks.4.norm2.weight",
559
+ "visual.blocks.5.norm1.weight",
560
+ "visual.blocks.5.norm2.weight",
561
+ "visual.blocks.6.norm1.weight",
562
+ "visual.blocks.6.norm2.weight",
563
+ "visual.blocks.7.norm1.weight",
564
+ "visual.blocks.7.norm2.weight",
565
+ "visual.blocks.8.norm1.weight",
566
+ "visual.blocks.8.norm2.weight",
567
+ "visual.blocks.9.norm1.weight",
568
+ "visual.blocks.9.norm2.weight",
569
+ "visual.blocks.10.norm1.weight",
570
+ "visual.blocks.10.norm2.weight",
571
+ "visual.blocks.11.norm1.weight",
572
+ "visual.blocks.11.norm2.weight",
573
+ "visual.blocks.12.norm1.weight",
574
+ "visual.blocks.12.norm2.weight",
575
+ "visual.blocks.13.norm1.weight",
576
+ "visual.blocks.13.norm2.weight",
577
+ "visual.blocks.14.norm1.weight",
578
+ "visual.blocks.14.norm2.weight",
579
+ "visual.blocks.15.norm1.weight",
580
+ "visual.blocks.15.norm2.weight",
581
+ "visual.blocks.16.norm1.weight",
582
+ "visual.blocks.16.norm2.weight",
583
+ "visual.blocks.17.norm1.weight",
584
+ "visual.blocks.17.norm2.weight",
585
+ "visual.blocks.18.norm1.weight",
586
+ "visual.blocks.18.norm2.weight",
587
+ "visual.blocks.19.norm1.weight",
588
+ "visual.blocks.19.norm2.weight",
589
+ "visual.blocks.20.norm1.weight",
590
+ "visual.blocks.20.norm2.weight",
591
+ "visual.blocks.21.norm1.weight",
592
+ "visual.blocks.21.norm2.weight",
593
+ "visual.blocks.22.norm1.weight",
594
+ "visual.blocks.22.norm2.weight",
595
+ "visual.blocks.23.norm1.weight",
596
+ "visual.blocks.23.norm2.weight",
597
+ "visual.deepstack_merger_list.0.norm.weight",
598
+ "visual.deepstack_merger_list.1.norm.weight",
599
+ "visual.deepstack_merger_list.2.norm.weight",
600
+ "visual.merger.norm.weight",
601
+ "visual.pos_embed.weight"
602
+ ],
603
+ "modules_to_not_use_matmul": [],
604
+ "non_blocking": false,
605
+ "quant_conv": false,
606
+ "quant_embedding": false,
607
+ "quant_method": "sdnq",
608
+ "quantization_device": "cuda",
609
+ "quantized_matmul_dtype": null,
610
+ "return_device": "cpu",
611
+ "sdnq_version": "0.2.1",
612
+ "svd_rank": 32,
613
+ "svd_steps": 8,
614
+ "use_dynamic_quantization": true,
615
+ "use_grad_ckpt": true,
616
+ "use_hadamard": true,
617
+ "use_quantized_matmul": true,
618
+ "use_quantized_matmul_conv": true,
619
+ "use_static_quantization": true,
620
+ "use_stochastic_rounding": false,
621
+ "use_svd": false,
622
+ "weights_dtype": "uint4"
623
+ },
624
+ "text_config": {
625
+ "attention_bias": false,
626
+ "attention_dropout": 0.0,
627
+ "bos_token_id": 151643,
628
+ "dtype": "bfloat16",
629
+ "eos_token_id": 151645,
630
+ "head_dim": 128,
631
+ "hidden_act": "silu",
632
+ "hidden_size": 2560,
633
+ "initializer_range": 0.02,
634
+ "intermediate_size": 9728,
635
+ "max_position_embeddings": 262144,
636
+ "model_type": "qwen3_vl_text",
637
+ "num_attention_heads": 32,
638
+ "num_hidden_layers": 36,
639
+ "num_key_value_heads": 8,
640
+ "pad_token_id": null,
641
+ "rms_norm_eps": 1e-06,
642
+ "rope_parameters": {
643
+ "mrope_interleaved": true,
644
+ "mrope_section": [
645
+ 24,
646
+ 20,
647
+ 20
648
+ ],
649
+ "rope_theta": 5000000,
650
+ "rope_type": "default"
651
+ },
652
+ "tie_word_embeddings": true,
653
+ "use_cache": true,
654
+ "vocab_size": 151936
655
+ },
656
+ "tie_word_embeddings": true,
657
+ "transformers_version": "5.13.0.dev0",
658
+ "video_token_id": 151656,
659
+ "vision_config": {
660
+ "deepstack_visual_indexes": [
661
+ 5,
662
+ 11,
663
+ 17
664
+ ],
665
+ "depth": 24,
666
+ "dtype": "bfloat16",
667
+ "hidden_act": "gelu_pytorch_tanh",
668
+ "hidden_size": 1024,
669
+ "in_channels": 3,
670
+ "initializer_range": 0.02,
671
+ "intermediate_size": 4096,
672
+ "model_type": "qwen3_vl_vision",
673
+ "num_heads": 16,
674
+ "num_position_embeddings": 2304,
675
+ "out_hidden_size": 2560,
676
+ "patch_size": 16,
677
+ "spatial_merge_size": 2,
678
+ "temporal_patch_size": 2
679
+ },
680
+ "vision_end_token_id": 151653,
681
+ "vision_start_token_id": 151652
682
+ }
text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5911bfcd683f68abb68bcdbdf90889aff4c98f163652627c1a1bebd8d30ecf3c
3
+ size 3062036496
text_encoder/quantization_config.json ADDED
@@ -0,0 +1,616 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_skip_keys": true,
3
+ "dequantize_fp32": false,
4
+ "dynamic_loss_threshold": -1,
5
+ "group_size": 0,
6
+ "hadamard_group_size": 256,
7
+ "is_integer": true,
8
+ "is_training": false,
9
+ "modules_dtype_dict": {
10
+ "uint4": [
11
+ "language_model.layers.0.mlp.down_proj.weight",
12
+ "language_model.layers.0.mlp.gate_proj.weight",
13
+ "language_model.layers.0.mlp.up_proj.weight",
14
+ "language_model.layers.0.self_attn.k_proj.weight",
15
+ "language_model.layers.0.self_attn.o_proj.weight",
16
+ "language_model.layers.0.self_attn.q_proj.weight",
17
+ "language_model.layers.0.self_attn.v_proj.weight",
18
+ "language_model.layers.1.mlp.down_proj.weight",
19
+ "language_model.layers.1.mlp.gate_proj.weight",
20
+ "language_model.layers.1.mlp.up_proj.weight",
21
+ "language_model.layers.1.self_attn.k_proj.weight",
22
+ "language_model.layers.1.self_attn.o_proj.weight",
23
+ "language_model.layers.1.self_attn.q_proj.weight",
24
+ "language_model.layers.1.self_attn.v_proj.weight",
25
+ "language_model.layers.2.mlp.down_proj.weight",
26
+ "language_model.layers.2.mlp.gate_proj.weight",
27
+ "language_model.layers.2.mlp.up_proj.weight",
28
+ "language_model.layers.2.self_attn.k_proj.weight",
29
+ "language_model.layers.2.self_attn.o_proj.weight",
30
+ "language_model.layers.2.self_attn.q_proj.weight",
31
+ "language_model.layers.2.self_attn.v_proj.weight",
32
+ "language_model.layers.3.mlp.down_proj.weight",
33
+ "language_model.layers.3.mlp.gate_proj.weight",
34
+ "language_model.layers.3.mlp.up_proj.weight",
35
+ "language_model.layers.3.self_attn.k_proj.weight",
36
+ "language_model.layers.3.self_attn.o_proj.weight",
37
+ "language_model.layers.3.self_attn.q_proj.weight",
38
+ "language_model.layers.3.self_attn.v_proj.weight",
39
+ "language_model.layers.4.mlp.down_proj.weight",
40
+ "language_model.layers.4.mlp.gate_proj.weight",
41
+ "language_model.layers.4.mlp.up_proj.weight",
42
+ "language_model.layers.4.self_attn.k_proj.weight",
43
+ "language_model.layers.4.self_attn.o_proj.weight",
44
+ "language_model.layers.4.self_attn.q_proj.weight",
45
+ "language_model.layers.4.self_attn.v_proj.weight",
46
+ "language_model.layers.5.mlp.down_proj.weight",
47
+ "language_model.layers.5.mlp.gate_proj.weight",
48
+ "language_model.layers.5.mlp.up_proj.weight",
49
+ "language_model.layers.5.self_attn.k_proj.weight",
50
+ "language_model.layers.5.self_attn.o_proj.weight",
51
+ "language_model.layers.5.self_attn.q_proj.weight",
52
+ "language_model.layers.5.self_attn.v_proj.weight",
53
+ "language_model.layers.6.mlp.down_proj.weight",
54
+ "language_model.layers.6.mlp.gate_proj.weight",
55
+ "language_model.layers.6.mlp.up_proj.weight",
56
+ "language_model.layers.6.self_attn.k_proj.weight",
57
+ "language_model.layers.6.self_attn.o_proj.weight",
58
+ "language_model.layers.6.self_attn.q_proj.weight",
59
+ "language_model.layers.6.self_attn.v_proj.weight",
60
+ "language_model.layers.7.mlp.down_proj.weight",
61
+ "language_model.layers.7.mlp.gate_proj.weight",
62
+ "language_model.layers.7.mlp.up_proj.weight",
63
+ "language_model.layers.7.self_attn.k_proj.weight",
64
+ "language_model.layers.7.self_attn.o_proj.weight",
65
+ "language_model.layers.7.self_attn.q_proj.weight",
66
+ "language_model.layers.7.self_attn.v_proj.weight",
67
+ "language_model.layers.8.mlp.down_proj.weight",
68
+ "language_model.layers.8.mlp.gate_proj.weight",
69
+ "language_model.layers.8.mlp.up_proj.weight",
70
+ "language_model.layers.8.self_attn.k_proj.weight",
71
+ "language_model.layers.8.self_attn.o_proj.weight",
72
+ "language_model.layers.8.self_attn.q_proj.weight",
73
+ "language_model.layers.8.self_attn.v_proj.weight",
74
+ "language_model.layers.9.mlp.down_proj.weight",
75
+ "language_model.layers.9.mlp.gate_proj.weight",
76
+ "language_model.layers.9.mlp.up_proj.weight",
77
+ "language_model.layers.9.self_attn.k_proj.weight",
78
+ "language_model.layers.9.self_attn.o_proj.weight",
79
+ "language_model.layers.9.self_attn.q_proj.weight",
80
+ "language_model.layers.9.self_attn.v_proj.weight",
81
+ "language_model.layers.10.mlp.down_proj.weight",
82
+ "language_model.layers.10.mlp.gate_proj.weight",
83
+ "language_model.layers.10.mlp.up_proj.weight",
84
+ "language_model.layers.10.self_attn.k_proj.weight",
85
+ "language_model.layers.10.self_attn.o_proj.weight",
86
+ "language_model.layers.10.self_attn.q_proj.weight",
87
+ "language_model.layers.10.self_attn.v_proj.weight",
88
+ "language_model.layers.11.mlp.down_proj.weight",
89
+ "language_model.layers.11.mlp.gate_proj.weight",
90
+ "language_model.layers.11.mlp.up_proj.weight",
91
+ "language_model.layers.11.self_attn.k_proj.weight",
92
+ "language_model.layers.11.self_attn.o_proj.weight",
93
+ "language_model.layers.11.self_attn.q_proj.weight",
94
+ "language_model.layers.11.self_attn.v_proj.weight",
95
+ "language_model.layers.12.mlp.down_proj.weight",
96
+ "language_model.layers.12.mlp.gate_proj.weight",
97
+ "language_model.layers.12.mlp.up_proj.weight",
98
+ "language_model.layers.12.self_attn.k_proj.weight",
99
+ "language_model.layers.12.self_attn.o_proj.weight",
100
+ "language_model.layers.12.self_attn.q_proj.weight",
101
+ "language_model.layers.12.self_attn.v_proj.weight",
102
+ "language_model.layers.13.mlp.down_proj.weight",
103
+ "language_model.layers.13.mlp.gate_proj.weight",
104
+ "language_model.layers.13.mlp.up_proj.weight",
105
+ "language_model.layers.13.self_attn.k_proj.weight",
106
+ "language_model.layers.13.self_attn.o_proj.weight",
107
+ "language_model.layers.13.self_attn.q_proj.weight",
108
+ "language_model.layers.13.self_attn.v_proj.weight",
109
+ "language_model.layers.14.mlp.down_proj.weight",
110
+ "language_model.layers.14.mlp.gate_proj.weight",
111
+ "language_model.layers.14.mlp.up_proj.weight",
112
+ "language_model.layers.14.self_attn.k_proj.weight",
113
+ "language_model.layers.14.self_attn.o_proj.weight",
114
+ "language_model.layers.14.self_attn.q_proj.weight",
115
+ "language_model.layers.14.self_attn.v_proj.weight",
116
+ "language_model.layers.15.mlp.down_proj.weight",
117
+ "language_model.layers.15.mlp.gate_proj.weight",
118
+ "language_model.layers.15.mlp.up_proj.weight",
119
+ "language_model.layers.15.self_attn.k_proj.weight",
120
+ "language_model.layers.15.self_attn.o_proj.weight",
121
+ "language_model.layers.15.self_attn.q_proj.weight",
122
+ "language_model.layers.15.self_attn.v_proj.weight",
123
+ "language_model.layers.16.mlp.down_proj.weight",
124
+ "language_model.layers.16.mlp.gate_proj.weight",
125
+ "language_model.layers.16.mlp.up_proj.weight",
126
+ "language_model.layers.16.self_attn.k_proj.weight",
127
+ "language_model.layers.16.self_attn.o_proj.weight",
128
+ "language_model.layers.16.self_attn.q_proj.weight",
129
+ "language_model.layers.16.self_attn.v_proj.weight",
130
+ "language_model.layers.17.mlp.down_proj.weight",
131
+ "language_model.layers.17.mlp.gate_proj.weight",
132
+ "language_model.layers.17.mlp.up_proj.weight",
133
+ "language_model.layers.17.self_attn.k_proj.weight",
134
+ "language_model.layers.17.self_attn.o_proj.weight",
135
+ "language_model.layers.17.self_attn.q_proj.weight",
136
+ "language_model.layers.17.self_attn.v_proj.weight",
137
+ "language_model.layers.18.mlp.down_proj.weight",
138
+ "language_model.layers.18.mlp.gate_proj.weight",
139
+ "language_model.layers.18.mlp.up_proj.weight",
140
+ "language_model.layers.18.self_attn.k_proj.weight",
141
+ "language_model.layers.18.self_attn.o_proj.weight",
142
+ "language_model.layers.18.self_attn.q_proj.weight",
143
+ "language_model.layers.18.self_attn.v_proj.weight",
144
+ "language_model.layers.19.mlp.down_proj.weight",
145
+ "language_model.layers.19.mlp.gate_proj.weight",
146
+ "language_model.layers.19.mlp.up_proj.weight",
147
+ "language_model.layers.19.self_attn.k_proj.weight",
148
+ "language_model.layers.19.self_attn.o_proj.weight",
149
+ "language_model.layers.19.self_attn.q_proj.weight",
150
+ "language_model.layers.19.self_attn.v_proj.weight",
151
+ "language_model.layers.20.mlp.down_proj.weight",
152
+ "language_model.layers.20.mlp.gate_proj.weight",
153
+ "language_model.layers.20.mlp.up_proj.weight",
154
+ "language_model.layers.20.self_attn.k_proj.weight",
155
+ "language_model.layers.20.self_attn.o_proj.weight",
156
+ "language_model.layers.20.self_attn.q_proj.weight",
157
+ "language_model.layers.20.self_attn.v_proj.weight",
158
+ "language_model.layers.21.mlp.down_proj.weight",
159
+ "language_model.layers.21.mlp.gate_proj.weight",
160
+ "language_model.layers.21.mlp.up_proj.weight",
161
+ "language_model.layers.21.self_attn.k_proj.weight",
162
+ "language_model.layers.21.self_attn.o_proj.weight",
163
+ "language_model.layers.21.self_attn.q_proj.weight",
164
+ "language_model.layers.21.self_attn.v_proj.weight",
165
+ "language_model.layers.22.mlp.down_proj.weight",
166
+ "language_model.layers.22.mlp.gate_proj.weight",
167
+ "language_model.layers.22.mlp.up_proj.weight",
168
+ "language_model.layers.22.self_attn.k_proj.weight",
169
+ "language_model.layers.22.self_attn.o_proj.weight",
170
+ "language_model.layers.22.self_attn.q_proj.weight",
171
+ "language_model.layers.22.self_attn.v_proj.weight",
172
+ "language_model.layers.23.mlp.down_proj.weight",
173
+ "language_model.layers.23.mlp.gate_proj.weight",
174
+ "language_model.layers.23.mlp.up_proj.weight",
175
+ "language_model.layers.23.self_attn.k_proj.weight",
176
+ "language_model.layers.23.self_attn.o_proj.weight",
177
+ "language_model.layers.23.self_attn.q_proj.weight",
178
+ "language_model.layers.23.self_attn.v_proj.weight",
179
+ "language_model.layers.24.mlp.down_proj.weight",
180
+ "language_model.layers.24.mlp.gate_proj.weight",
181
+ "language_model.layers.24.mlp.up_proj.weight",
182
+ "language_model.layers.24.self_attn.k_proj.weight",
183
+ "language_model.layers.24.self_attn.o_proj.weight",
184
+ "language_model.layers.24.self_attn.q_proj.weight",
185
+ "language_model.layers.24.self_attn.v_proj.weight",
186
+ "language_model.layers.25.mlp.down_proj.weight",
187
+ "language_model.layers.25.mlp.gate_proj.weight",
188
+ "language_model.layers.25.mlp.up_proj.weight",
189
+ "language_model.layers.25.self_attn.k_proj.weight",
190
+ "language_model.layers.25.self_attn.o_proj.weight",
191
+ "language_model.layers.25.self_attn.q_proj.weight",
192
+ "language_model.layers.25.self_attn.v_proj.weight",
193
+ "language_model.layers.26.mlp.down_proj.weight",
194
+ "language_model.layers.26.mlp.gate_proj.weight",
195
+ "language_model.layers.26.mlp.up_proj.weight",
196
+ "language_model.layers.26.self_attn.k_proj.weight",
197
+ "language_model.layers.26.self_attn.o_proj.weight",
198
+ "language_model.layers.26.self_attn.q_proj.weight",
199
+ "language_model.layers.26.self_attn.v_proj.weight",
200
+ "language_model.layers.27.mlp.down_proj.weight",
201
+ "language_model.layers.27.mlp.gate_proj.weight",
202
+ "language_model.layers.27.mlp.up_proj.weight",
203
+ "language_model.layers.27.self_attn.k_proj.weight",
204
+ "language_model.layers.27.self_attn.o_proj.weight",
205
+ "language_model.layers.27.self_attn.q_proj.weight",
206
+ "language_model.layers.27.self_attn.v_proj.weight",
207
+ "language_model.layers.28.mlp.down_proj.weight",
208
+ "language_model.layers.28.mlp.gate_proj.weight",
209
+ "language_model.layers.28.mlp.up_proj.weight",
210
+ "language_model.layers.28.self_attn.k_proj.weight",
211
+ "language_model.layers.28.self_attn.o_proj.weight",
212
+ "language_model.layers.28.self_attn.q_proj.weight",
213
+ "language_model.layers.28.self_attn.v_proj.weight",
214
+ "language_model.layers.29.mlp.down_proj.weight",
215
+ "language_model.layers.29.mlp.gate_proj.weight",
216
+ "language_model.layers.29.mlp.up_proj.weight",
217
+ "language_model.layers.29.self_attn.k_proj.weight",
218
+ "language_model.layers.29.self_attn.o_proj.weight",
219
+ "language_model.layers.29.self_attn.q_proj.weight",
220
+ "language_model.layers.29.self_attn.v_proj.weight",
221
+ "language_model.layers.30.mlp.down_proj.weight",
222
+ "language_model.layers.30.mlp.gate_proj.weight",
223
+ "language_model.layers.30.mlp.up_proj.weight",
224
+ "language_model.layers.30.self_attn.k_proj.weight",
225
+ "language_model.layers.30.self_attn.o_proj.weight",
226
+ "language_model.layers.30.self_attn.q_proj.weight",
227
+ "language_model.layers.30.self_attn.v_proj.weight",
228
+ "language_model.layers.31.mlp.down_proj.weight",
229
+ "language_model.layers.31.mlp.gate_proj.weight",
230
+ "language_model.layers.31.mlp.up_proj.weight",
231
+ "language_model.layers.31.self_attn.k_proj.weight",
232
+ "language_model.layers.31.self_attn.o_proj.weight",
233
+ "language_model.layers.31.self_attn.q_proj.weight",
234
+ "language_model.layers.31.self_attn.v_proj.weight",
235
+ "language_model.layers.32.mlp.down_proj.weight",
236
+ "language_model.layers.32.mlp.gate_proj.weight",
237
+ "language_model.layers.32.mlp.up_proj.weight",
238
+ "language_model.layers.32.self_attn.k_proj.weight",
239
+ "language_model.layers.32.self_attn.o_proj.weight",
240
+ "language_model.layers.32.self_attn.q_proj.weight",
241
+ "language_model.layers.32.self_attn.v_proj.weight",
242
+ "language_model.layers.33.mlp.down_proj.weight",
243
+ "language_model.layers.33.mlp.gate_proj.weight",
244
+ "language_model.layers.33.mlp.up_proj.weight",
245
+ "language_model.layers.33.self_attn.k_proj.weight",
246
+ "language_model.layers.33.self_attn.o_proj.weight",
247
+ "language_model.layers.33.self_attn.q_proj.weight",
248
+ "language_model.layers.33.self_attn.v_proj.weight",
249
+ "language_model.layers.34.mlp.down_proj.weight",
250
+ "language_model.layers.34.mlp.gate_proj.weight",
251
+ "language_model.layers.34.mlp.up_proj.weight",
252
+ "language_model.layers.34.self_attn.k_proj.weight",
253
+ "language_model.layers.34.self_attn.o_proj.weight",
254
+ "language_model.layers.34.self_attn.q_proj.weight",
255
+ "language_model.layers.34.self_attn.v_proj.weight",
256
+ "language_model.layers.35.mlp.down_proj.weight",
257
+ "language_model.layers.35.mlp.gate_proj.weight",
258
+ "language_model.layers.35.mlp.up_proj.weight",
259
+ "language_model.layers.35.self_attn.k_proj.weight",
260
+ "language_model.layers.35.self_attn.o_proj.weight",
261
+ "language_model.layers.35.self_attn.q_proj.weight",
262
+ "language_model.layers.35.self_attn.v_proj.weight",
263
+ "visual.blocks.0.attn.proj.weight",
264
+ "visual.blocks.0.attn.qkv.weight",
265
+ "visual.blocks.0.mlp.linear_fc1.weight",
266
+ "visual.blocks.0.mlp.linear_fc2.weight",
267
+ "visual.blocks.1.attn.proj.weight",
268
+ "visual.blocks.1.attn.qkv.weight",
269
+ "visual.blocks.1.mlp.linear_fc1.weight",
270
+ "visual.blocks.1.mlp.linear_fc2.weight",
271
+ "visual.blocks.2.attn.proj.weight",
272
+ "visual.blocks.2.attn.qkv.weight",
273
+ "visual.blocks.2.mlp.linear_fc1.weight",
274
+ "visual.blocks.2.mlp.linear_fc2.weight",
275
+ "visual.blocks.3.attn.proj.weight",
276
+ "visual.blocks.3.attn.qkv.weight",
277
+ "visual.blocks.3.mlp.linear_fc1.weight",
278
+ "visual.blocks.3.mlp.linear_fc2.weight",
279
+ "visual.blocks.4.attn.proj.weight",
280
+ "visual.blocks.4.attn.qkv.weight",
281
+ "visual.blocks.4.mlp.linear_fc1.weight",
282
+ "visual.blocks.4.mlp.linear_fc2.weight",
283
+ "visual.blocks.5.attn.proj.weight",
284
+ "visual.blocks.5.attn.qkv.weight",
285
+ "visual.blocks.5.mlp.linear_fc1.weight",
286
+ "visual.blocks.5.mlp.linear_fc2.weight",
287
+ "visual.blocks.6.attn.proj.weight",
288
+ "visual.blocks.6.attn.qkv.weight",
289
+ "visual.blocks.6.mlp.linear_fc1.weight",
290
+ "visual.blocks.6.mlp.linear_fc2.weight",
291
+ "visual.blocks.7.attn.proj.weight",
292
+ "visual.blocks.7.attn.qkv.weight",
293
+ "visual.blocks.7.mlp.linear_fc1.weight",
294
+ "visual.blocks.7.mlp.linear_fc2.weight",
295
+ "visual.blocks.8.attn.proj.weight",
296
+ "visual.blocks.8.attn.qkv.weight",
297
+ "visual.blocks.8.mlp.linear_fc1.weight",
298
+ "visual.blocks.8.mlp.linear_fc2.weight",
299
+ "visual.blocks.9.attn.proj.weight",
300
+ "visual.blocks.9.attn.qkv.weight",
301
+ "visual.blocks.9.mlp.linear_fc1.weight",
302
+ "visual.blocks.9.mlp.linear_fc2.weight",
303
+ "visual.blocks.10.attn.proj.weight",
304
+ "visual.blocks.10.attn.qkv.weight",
305
+ "visual.blocks.10.mlp.linear_fc1.weight",
306
+ "visual.blocks.10.mlp.linear_fc2.weight",
307
+ "visual.blocks.11.attn.proj.weight",
308
+ "visual.blocks.11.attn.qkv.weight",
309
+ "visual.blocks.11.mlp.linear_fc1.weight",
310
+ "visual.blocks.11.mlp.linear_fc2.weight",
311
+ "visual.blocks.12.attn.proj.weight",
312
+ "visual.blocks.12.attn.qkv.weight",
313
+ "visual.blocks.12.mlp.linear_fc1.weight",
314
+ "visual.blocks.12.mlp.linear_fc2.weight",
315
+ "visual.blocks.13.attn.proj.weight",
316
+ "visual.blocks.13.attn.qkv.weight",
317
+ "visual.blocks.13.mlp.linear_fc1.weight",
318
+ "visual.blocks.13.mlp.linear_fc2.weight",
319
+ "visual.blocks.14.attn.proj.weight",
320
+ "visual.blocks.14.attn.qkv.weight",
321
+ "visual.blocks.14.mlp.linear_fc1.weight",
322
+ "visual.blocks.14.mlp.linear_fc2.weight",
323
+ "visual.blocks.15.attn.proj.weight",
324
+ "visual.blocks.15.attn.qkv.weight",
325
+ "visual.blocks.15.mlp.linear_fc1.weight",
326
+ "visual.blocks.15.mlp.linear_fc2.weight",
327
+ "visual.blocks.16.attn.proj.weight",
328
+ "visual.blocks.16.attn.qkv.weight",
329
+ "visual.blocks.16.mlp.linear_fc1.weight",
330
+ "visual.blocks.16.mlp.linear_fc2.weight",
331
+ "visual.blocks.17.attn.proj.weight",
332
+ "visual.blocks.17.attn.qkv.weight",
333
+ "visual.blocks.17.mlp.linear_fc1.weight",
334
+ "visual.blocks.17.mlp.linear_fc2.weight",
335
+ "visual.blocks.18.attn.proj.weight",
336
+ "visual.blocks.18.attn.qkv.weight",
337
+ "visual.blocks.18.mlp.linear_fc1.weight",
338
+ "visual.blocks.18.mlp.linear_fc2.weight",
339
+ "visual.blocks.19.attn.proj.weight",
340
+ "visual.blocks.19.attn.qkv.weight",
341
+ "visual.blocks.19.mlp.linear_fc1.weight",
342
+ "visual.blocks.19.mlp.linear_fc2.weight",
343
+ "visual.blocks.20.attn.proj.weight",
344
+ "visual.blocks.20.attn.qkv.weight",
345
+ "visual.blocks.20.mlp.linear_fc1.weight",
346
+ "visual.blocks.20.mlp.linear_fc2.weight",
347
+ "visual.blocks.21.attn.proj.weight",
348
+ "visual.blocks.21.attn.qkv.weight",
349
+ "visual.blocks.21.mlp.linear_fc1.weight",
350
+ "visual.blocks.21.mlp.linear_fc2.weight",
351
+ "visual.blocks.22.attn.proj.weight",
352
+ "visual.blocks.22.attn.qkv.weight",
353
+ "visual.blocks.22.mlp.linear_fc1.weight",
354
+ "visual.blocks.22.mlp.linear_fc2.weight",
355
+ "visual.blocks.23.attn.proj.weight",
356
+ "visual.blocks.23.attn.qkv.weight",
357
+ "visual.blocks.23.mlp.linear_fc1.weight",
358
+ "visual.blocks.23.mlp.linear_fc2.weight",
359
+ "visual.deepstack_merger_list.0.linear_fc1.weight",
360
+ "visual.deepstack_merger_list.0.linear_fc2.weight",
361
+ "visual.deepstack_merger_list.1.linear_fc1.weight",
362
+ "visual.deepstack_merger_list.1.linear_fc2.weight",
363
+ "visual.deepstack_merger_list.2.linear_fc1.weight",
364
+ "visual.deepstack_merger_list.2.linear_fc2.weight",
365
+ "visual.merger.linear_fc1.weight",
366
+ "visual.merger.linear_fc2.weight"
367
+ ]
368
+ },
369
+ "modules_quant_config": {},
370
+ "modules_to_not_convert": [
371
+ ".txt_out",
372
+ ".time_embed",
373
+ ".vid_out",
374
+ ".vid_in",
375
+ ".txt_in",
376
+ ".img_in",
377
+ ".emb_out",
378
+ ".norm_out",
379
+ "multi_modal_projector",
380
+ ".img_out",
381
+ ".t_embedder",
382
+ ".final_layer",
383
+ ".emb_in",
384
+ "patch_embedding",
385
+ "language_model.norm",
386
+ ".proj_out",
387
+ "time_text_embed",
388
+ "patch_embed",
389
+ ".x_embedder",
390
+ "patch_emb",
391
+ "wte",
392
+ ".y_embedder",
393
+ ".condition_embedder",
394
+ ".context_embedder",
395
+ "lm_head",
396
+ "language_model.embed_tokens.weight",
397
+ "language_model.layers.0.input_layernorm.weight",
398
+ "language_model.layers.0.post_attention_layernorm.weight",
399
+ "language_model.layers.0.self_attn.k_norm.weight",
400
+ "language_model.layers.0.self_attn.q_norm.weight",
401
+ "language_model.layers.1.input_layernorm.weight",
402
+ "language_model.layers.1.post_attention_layernorm.weight",
403
+ "language_model.layers.1.self_attn.k_norm.weight",
404
+ "language_model.layers.1.self_attn.q_norm.weight",
405
+ "language_model.layers.2.input_layernorm.weight",
406
+ "language_model.layers.2.post_attention_layernorm.weight",
407
+ "language_model.layers.2.self_attn.k_norm.weight",
408
+ "language_model.layers.2.self_attn.q_norm.weight",
409
+ "language_model.layers.3.input_layernorm.weight",
410
+ "language_model.layers.3.post_attention_layernorm.weight",
411
+ "language_model.layers.3.self_attn.k_norm.weight",
412
+ "language_model.layers.3.self_attn.q_norm.weight",
413
+ "language_model.layers.4.input_layernorm.weight",
414
+ "language_model.layers.4.post_attention_layernorm.weight",
415
+ "language_model.layers.4.self_attn.k_norm.weight",
416
+ "language_model.layers.4.self_attn.q_norm.weight",
417
+ "language_model.layers.5.input_layernorm.weight",
418
+ "language_model.layers.5.post_attention_layernorm.weight",
419
+ "language_model.layers.5.self_attn.k_norm.weight",
420
+ "language_model.layers.5.self_attn.q_norm.weight",
421
+ "language_model.layers.6.input_layernorm.weight",
422
+ "language_model.layers.6.post_attention_layernorm.weight",
423
+ "language_model.layers.6.self_attn.k_norm.weight",
424
+ "language_model.layers.6.self_attn.q_norm.weight",
425
+ "language_model.layers.7.input_layernorm.weight",
426
+ "language_model.layers.7.post_attention_layernorm.weight",
427
+ "language_model.layers.7.self_attn.k_norm.weight",
428
+ "language_model.layers.7.self_attn.q_norm.weight",
429
+ "language_model.layers.8.input_layernorm.weight",
430
+ "language_model.layers.8.post_attention_layernorm.weight",
431
+ "language_model.layers.8.self_attn.k_norm.weight",
432
+ "language_model.layers.8.self_attn.q_norm.weight",
433
+ "language_model.layers.9.input_layernorm.weight",
434
+ "language_model.layers.9.post_attention_layernorm.weight",
435
+ "language_model.layers.9.self_attn.k_norm.weight",
436
+ "language_model.layers.9.self_attn.q_norm.weight",
437
+ "language_model.layers.10.input_layernorm.weight",
438
+ "language_model.layers.10.post_attention_layernorm.weight",
439
+ "language_model.layers.10.self_attn.k_norm.weight",
440
+ "language_model.layers.10.self_attn.q_norm.weight",
441
+ "language_model.layers.11.input_layernorm.weight",
442
+ "language_model.layers.11.post_attention_layernorm.weight",
443
+ "language_model.layers.11.self_attn.k_norm.weight",
444
+ "language_model.layers.11.self_attn.q_norm.weight",
445
+ "language_model.layers.12.input_layernorm.weight",
446
+ "language_model.layers.12.post_attention_layernorm.weight",
447
+ "language_model.layers.12.self_attn.k_norm.weight",
448
+ "language_model.layers.12.self_attn.q_norm.weight",
449
+ "language_model.layers.13.input_layernorm.weight",
450
+ "language_model.layers.13.post_attention_layernorm.weight",
451
+ "language_model.layers.13.self_attn.k_norm.weight",
452
+ "language_model.layers.13.self_attn.q_norm.weight",
453
+ "language_model.layers.14.input_layernorm.weight",
454
+ "language_model.layers.14.post_attention_layernorm.weight",
455
+ "language_model.layers.14.self_attn.k_norm.weight",
456
+ "language_model.layers.14.self_attn.q_norm.weight",
457
+ "language_model.layers.15.input_layernorm.weight",
458
+ "language_model.layers.15.post_attention_layernorm.weight",
459
+ "language_model.layers.15.self_attn.k_norm.weight",
460
+ "language_model.layers.15.self_attn.q_norm.weight",
461
+ "language_model.layers.16.input_layernorm.weight",
462
+ "language_model.layers.16.post_attention_layernorm.weight",
463
+ "language_model.layers.16.self_attn.k_norm.weight",
464
+ "language_model.layers.16.self_attn.q_norm.weight",
465
+ "language_model.layers.17.input_layernorm.weight",
466
+ "language_model.layers.17.post_attention_layernorm.weight",
467
+ "language_model.layers.17.self_attn.k_norm.weight",
468
+ "language_model.layers.17.self_attn.q_norm.weight",
469
+ "language_model.layers.18.input_layernorm.weight",
470
+ "language_model.layers.18.post_attention_layernorm.weight",
471
+ "language_model.layers.18.self_attn.k_norm.weight",
472
+ "language_model.layers.18.self_attn.q_norm.weight",
473
+ "language_model.layers.19.input_layernorm.weight",
474
+ "language_model.layers.19.post_attention_layernorm.weight",
475
+ "language_model.layers.19.self_attn.k_norm.weight",
476
+ "language_model.layers.19.self_attn.q_norm.weight",
477
+ "language_model.layers.20.input_layernorm.weight",
478
+ "language_model.layers.20.post_attention_layernorm.weight",
479
+ "language_model.layers.20.self_attn.k_norm.weight",
480
+ "language_model.layers.20.self_attn.q_norm.weight",
481
+ "language_model.layers.21.input_layernorm.weight",
482
+ "language_model.layers.21.post_attention_layernorm.weight",
483
+ "language_model.layers.21.self_attn.k_norm.weight",
484
+ "language_model.layers.21.self_attn.q_norm.weight",
485
+ "language_model.layers.22.input_layernorm.weight",
486
+ "language_model.layers.22.post_attention_layernorm.weight",
487
+ "language_model.layers.22.self_attn.k_norm.weight",
488
+ "language_model.layers.22.self_attn.q_norm.weight",
489
+ "language_model.layers.23.input_layernorm.weight",
490
+ "language_model.layers.23.post_attention_layernorm.weight",
491
+ "language_model.layers.23.self_attn.k_norm.weight",
492
+ "language_model.layers.23.self_attn.q_norm.weight",
493
+ "language_model.layers.24.input_layernorm.weight",
494
+ "language_model.layers.24.post_attention_layernorm.weight",
495
+ "language_model.layers.24.self_attn.k_norm.weight",
496
+ "language_model.layers.24.self_attn.q_norm.weight",
497
+ "language_model.layers.25.input_layernorm.weight",
498
+ "language_model.layers.25.post_attention_layernorm.weight",
499
+ "language_model.layers.25.self_attn.k_norm.weight",
500
+ "language_model.layers.25.self_attn.q_norm.weight",
501
+ "language_model.layers.26.input_layernorm.weight",
502
+ "language_model.layers.26.post_attention_layernorm.weight",
503
+ "language_model.layers.26.self_attn.k_norm.weight",
504
+ "language_model.layers.26.self_attn.q_norm.weight",
505
+ "language_model.layers.27.input_layernorm.weight",
506
+ "language_model.layers.27.post_attention_layernorm.weight",
507
+ "language_model.layers.27.self_attn.k_norm.weight",
508
+ "language_model.layers.27.self_attn.q_norm.weight",
509
+ "language_model.layers.28.input_layernorm.weight",
510
+ "language_model.layers.28.post_attention_layernorm.weight",
511
+ "language_model.layers.28.self_attn.k_norm.weight",
512
+ "language_model.layers.28.self_attn.q_norm.weight",
513
+ "language_model.layers.29.input_layernorm.weight",
514
+ "language_model.layers.29.post_attention_layernorm.weight",
515
+ "language_model.layers.29.self_attn.k_norm.weight",
516
+ "language_model.layers.29.self_attn.q_norm.weight",
517
+ "language_model.layers.30.input_layernorm.weight",
518
+ "language_model.layers.30.post_attention_layernorm.weight",
519
+ "language_model.layers.30.self_attn.k_norm.weight",
520
+ "language_model.layers.30.self_attn.q_norm.weight",
521
+ "language_model.layers.31.input_layernorm.weight",
522
+ "language_model.layers.31.post_attention_layernorm.weight",
523
+ "language_model.layers.31.self_attn.k_norm.weight",
524
+ "language_model.layers.31.self_attn.q_norm.weight",
525
+ "language_model.layers.32.input_layernorm.weight",
526
+ "language_model.layers.32.post_attention_layernorm.weight",
527
+ "language_model.layers.32.self_attn.k_norm.weight",
528
+ "language_model.layers.32.self_attn.q_norm.weight",
529
+ "language_model.layers.33.input_layernorm.weight",
530
+ "language_model.layers.33.post_attention_layernorm.weight",
531
+ "language_model.layers.33.self_attn.k_norm.weight",
532
+ "language_model.layers.33.self_attn.q_norm.weight",
533
+ "language_model.layers.34.input_layernorm.weight",
534
+ "language_model.layers.34.post_attention_layernorm.weight",
535
+ "language_model.layers.34.self_attn.k_norm.weight",
536
+ "language_model.layers.34.self_attn.q_norm.weight",
537
+ "language_model.layers.35.input_layernorm.weight",
538
+ "language_model.layers.35.post_attention_layernorm.weight",
539
+ "language_model.layers.35.self_attn.k_norm.weight",
540
+ "language_model.layers.35.self_attn.q_norm.weight",
541
+ "language_model.norm.weight",
542
+ "visual.blocks.0.norm1.weight",
543
+ "visual.blocks.0.norm2.weight",
544
+ "visual.blocks.1.norm1.weight",
545
+ "visual.blocks.1.norm2.weight",
546
+ "visual.blocks.2.norm1.weight",
547
+ "visual.blocks.2.norm2.weight",
548
+ "visual.blocks.3.norm1.weight",
549
+ "visual.blocks.3.norm2.weight",
550
+ "visual.blocks.4.norm1.weight",
551
+ "visual.blocks.4.norm2.weight",
552
+ "visual.blocks.5.norm1.weight",
553
+ "visual.blocks.5.norm2.weight",
554
+ "visual.blocks.6.norm1.weight",
555
+ "visual.blocks.6.norm2.weight",
556
+ "visual.blocks.7.norm1.weight",
557
+ "visual.blocks.7.norm2.weight",
558
+ "visual.blocks.8.norm1.weight",
559
+ "visual.blocks.8.norm2.weight",
560
+ "visual.blocks.9.norm1.weight",
561
+ "visual.blocks.9.norm2.weight",
562
+ "visual.blocks.10.norm1.weight",
563
+ "visual.blocks.10.norm2.weight",
564
+ "visual.blocks.11.norm1.weight",
565
+ "visual.blocks.11.norm2.weight",
566
+ "visual.blocks.12.norm1.weight",
567
+ "visual.blocks.12.norm2.weight",
568
+ "visual.blocks.13.norm1.weight",
569
+ "visual.blocks.13.norm2.weight",
570
+ "visual.blocks.14.norm1.weight",
571
+ "visual.blocks.14.norm2.weight",
572
+ "visual.blocks.15.norm1.weight",
573
+ "visual.blocks.15.norm2.weight",
574
+ "visual.blocks.16.norm1.weight",
575
+ "visual.blocks.16.norm2.weight",
576
+ "visual.blocks.17.norm1.weight",
577
+ "visual.blocks.17.norm2.weight",
578
+ "visual.blocks.18.norm1.weight",
579
+ "visual.blocks.18.norm2.weight",
580
+ "visual.blocks.19.norm1.weight",
581
+ "visual.blocks.19.norm2.weight",
582
+ "visual.blocks.20.norm1.weight",
583
+ "visual.blocks.20.norm2.weight",
584
+ "visual.blocks.21.norm1.weight",
585
+ "visual.blocks.21.norm2.weight",
586
+ "visual.blocks.22.norm1.weight",
587
+ "visual.blocks.22.norm2.weight",
588
+ "visual.blocks.23.norm1.weight",
589
+ "visual.blocks.23.norm2.weight",
590
+ "visual.deepstack_merger_list.0.norm.weight",
591
+ "visual.deepstack_merger_list.1.norm.weight",
592
+ "visual.deepstack_merger_list.2.norm.weight",
593
+ "visual.merger.norm.weight",
594
+ "visual.pos_embed.weight"
595
+ ],
596
+ "modules_to_not_use_matmul": [],
597
+ "non_blocking": false,
598
+ "quant_conv": false,
599
+ "quant_embedding": false,
600
+ "quant_method": "sdnq",
601
+ "quantization_device": "cuda",
602
+ "quantized_matmul_dtype": null,
603
+ "return_device": "cpu",
604
+ "sdnq_version": "0.2.1",
605
+ "svd_rank": 32,
606
+ "svd_steps": 8,
607
+ "use_dynamic_quantization": true,
608
+ "use_grad_ckpt": true,
609
+ "use_hadamard": true,
610
+ "use_quantized_matmul": true,
611
+ "use_quantized_matmul_conv": true,
612
+ "use_static_quantization": true,
613
+ "use_stochastic_rounding": false,
614
+ "use_svd": false,
615
+ "weights_dtype": "uint4"
616
+ }
tokenizer/chat_template.jinja ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {%- if messages[0].content is string %}
5
+ {{- messages[0].content }}
6
+ {%- else %}
7
+ {%- for content in messages[0].content %}
8
+ {%- if 'text' in content %}
9
+ {{- content.text }}
10
+ {%- endif %}
11
+ {%- endfor %}
12
+ {%- endif %}
13
+ {{- '\n\n' }}
14
+ {%- endif %}
15
+ {{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
16
+ {%- for tool in tools %}
17
+ {{- "\n" }}
18
+ {{- tool | tojson }}
19
+ {%- endfor %}
20
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
21
+ {%- else %}
22
+ {%- if messages[0].role == 'system' %}
23
+ {{- '<|im_start|>system\n' }}
24
+ {%- if messages[0].content is string %}
25
+ {{- messages[0].content }}
26
+ {%- else %}
27
+ {%- for content in messages[0].content %}
28
+ {%- if 'text' in content %}
29
+ {{- content.text }}
30
+ {%- endif %}
31
+ {%- endfor %}
32
+ {%- endif %}
33
+ {{- '<|im_end|>\n' }}
34
+ {%- endif %}
35
+ {%- endif %}
36
+ {%- set image_count = namespace(value=0) %}
37
+ {%- set video_count = namespace(value=0) %}
38
+ {%- for message in messages %}
39
+ {%- if message.role == "user" %}
40
+ {{- '<|im_start|>' + message.role + '\n' }}
41
+ {%- if message.content is string %}
42
+ {{- message.content }}
43
+ {%- else %}
44
+ {%- for content in message.content %}
45
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
46
+ {%- set image_count.value = image_count.value + 1 %}
47
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
48
+ <|vision_start|><|image_pad|><|vision_end|>
49
+ {%- elif content.type == 'video' or 'video' in content %}
50
+ {%- set video_count.value = video_count.value + 1 %}
51
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
52
+ <|vision_start|><|video_pad|><|vision_end|>
53
+ {%- elif 'text' in content %}
54
+ {{- content.text }}
55
+ {%- endif %}
56
+ {%- endfor %}
57
+ {%- endif %}
58
+ {{- '<|im_end|>\n' }}
59
+ {%- elif message.role == "assistant" %}
60
+ {{- '<|im_start|>' + message.role + '\n' }}
61
+ {%- if message.content is string %}
62
+ {{- message.content }}
63
+ {%- else %}
64
+ {%- for content_item in message.content %}
65
+ {%- if 'text' in content_item %}
66
+ {{- content_item.text }}
67
+ {%- endif %}
68
+ {%- endfor %}
69
+ {%- endif %}
70
+ {%- if message.tool_calls %}
71
+ {%- for tool_call in message.tool_calls %}
72
+ {%- if (loop.first and message.content) or (not loop.first) %}
73
+ {{- '\n' }}
74
+ {%- endif %}
75
+ {%- if tool_call.function %}
76
+ {%- set tool_call = tool_call.function %}
77
+ {%- endif %}
78
+ {{- '<tool_call>\n{"name": "' }}
79
+ {{- tool_call.name }}
80
+ {{- '", "arguments": ' }}
81
+ {%- if tool_call.arguments is string %}
82
+ {{- tool_call.arguments }}
83
+ {%- else %}
84
+ {{- tool_call.arguments | tojson }}
85
+ {%- endif %}
86
+ {{- '}\n</tool_call>' }}
87
+ {%- endfor %}
88
+ {%- endif %}
89
+ {{- '<|im_end|>\n' }}
90
+ {%- elif message.role == "tool" %}
91
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
92
+ {{- '<|im_start|>user' }}
93
+ {%- endif %}
94
+ {{- '\n<tool_response>\n' }}
95
+ {%- if message.content is string %}
96
+ {{- message.content }}
97
+ {%- else %}
98
+ {%- for content in message.content %}
99
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
100
+ {%- set image_count.value = image_count.value + 1 %}
101
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
102
+ <|vision_start|><|image_pad|><|vision_end|>
103
+ {%- elif content.type == 'video' or 'video' in content %}
104
+ {%- set video_count.value = video_count.value + 1 %}
105
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
106
+ <|vision_start|><|video_pad|><|vision_end|>
107
+ {%- elif 'text' in content %}
108
+ {{- content.text }}
109
+ {%- endif %}
110
+ {%- endfor %}
111
+ {%- endif %}
112
+ {{- '\n</tool_response>' }}
113
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
114
+ {{- '<|im_end|>\n' }}
115
+ {%- endif %}
116
+ {%- endif %}
117
+ {%- endfor %}
118
+ {%- if add_generation_prompt %}
119
+ {{- '<|im_start|>assistant\n' }}
120
+ {%- endif %}
tokenizer/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be75606093db2094d7cd20f3c2f385c212750648bd6ea4fb2bf507a6a4c55506
3
+ size 11422650
tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|im_start|>",
10
+ "<|im_end|>",
11
+ "<|object_ref_start|>",
12
+ "<|object_ref_end|>",
13
+ "<|box_start|>",
14
+ "<|box_end|>",
15
+ "<|quad_start|>",
16
+ "<|quad_end|>",
17
+ "<|vision_start|>",
18
+ "<|vision_end|>",
19
+ "<|vision_pad|>",
20
+ "<|image_pad|>",
21
+ "<|video_pad|>"
22
+ ],
23
+ "is_local": true,
24
+ "local_files_only": false,
25
+ "model_max_length": 262144,
26
+ "pad_token": "<|endoftext|>",
27
+ "split_special_tokens": false,
28
+ "tokenizer_class": "Qwen2Tokenizer",
29
+ "unk_token": null
30
+ }
transformer/config.json ADDED
@@ -0,0 +1,343 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Transformer2DModel",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "CalamitousFelicitousness/Krea-2-Base-Diffusers",
5
+ "bias": false,
6
+ "channels": 16,
7
+ "features": 6144,
8
+ "heads": 48,
9
+ "is_distilled": false,
10
+ "kvheads": 12,
11
+ "layers": 28,
12
+ "multiplier": 4,
13
+ "patch": 2,
14
+ "quantization_config": {
15
+ "add_skip_keys": true,
16
+ "dequantize_fp32": false,
17
+ "dynamic_loss_threshold": -1,
18
+ "group_size": 0,
19
+ "hadamard_group_size": 256,
20
+ "is_integer": true,
21
+ "is_training": false,
22
+ "modules_dtype_dict": {
23
+ "uint4": [
24
+ "blocks.0.attn.gate.weight",
25
+ "blocks.0.attn.wk.weight",
26
+ "blocks.0.attn.wo.weight",
27
+ "blocks.0.attn.wq.weight",
28
+ "blocks.0.attn.wv.weight",
29
+ "blocks.0.mlp.down.weight",
30
+ "blocks.0.mlp.gate.weight",
31
+ "blocks.0.mlp.up.weight",
32
+ "blocks.1.attn.gate.weight",
33
+ "blocks.1.attn.wk.weight",
34
+ "blocks.1.attn.wo.weight",
35
+ "blocks.1.attn.wq.weight",
36
+ "blocks.1.attn.wv.weight",
37
+ "blocks.1.mlp.down.weight",
38
+ "blocks.1.mlp.gate.weight",
39
+ "blocks.1.mlp.up.weight",
40
+ "blocks.2.attn.gate.weight",
41
+ "blocks.2.attn.wk.weight",
42
+ "blocks.2.attn.wo.weight",
43
+ "blocks.2.attn.wq.weight",
44
+ "blocks.2.attn.wv.weight",
45
+ "blocks.2.mlp.down.weight",
46
+ "blocks.2.mlp.gate.weight",
47
+ "blocks.2.mlp.up.weight",
48
+ "blocks.3.attn.gate.weight",
49
+ "blocks.3.attn.wk.weight",
50
+ "blocks.3.attn.wo.weight",
51
+ "blocks.3.attn.wq.weight",
52
+ "blocks.3.attn.wv.weight",
53
+ "blocks.3.mlp.down.weight",
54
+ "blocks.3.mlp.gate.weight",
55
+ "blocks.3.mlp.up.weight",
56
+ "blocks.4.attn.gate.weight",
57
+ "blocks.4.attn.wk.weight",
58
+ "blocks.4.attn.wo.weight",
59
+ "blocks.4.attn.wq.weight",
60
+ "blocks.4.attn.wv.weight",
61
+ "blocks.4.mlp.down.weight",
62
+ "blocks.4.mlp.gate.weight",
63
+ "blocks.4.mlp.up.weight",
64
+ "blocks.5.attn.gate.weight",
65
+ "blocks.5.attn.wk.weight",
66
+ "blocks.5.attn.wo.weight",
67
+ "blocks.5.attn.wq.weight",
68
+ "blocks.5.attn.wv.weight",
69
+ "blocks.5.mlp.gate.weight",
70
+ "blocks.10.attn.gate.weight",
71
+ "blocks.10.attn.wk.weight",
72
+ "blocks.10.attn.wo.weight",
73
+ "blocks.10.attn.wq.weight",
74
+ "blocks.10.attn.wv.weight",
75
+ "blocks.10.mlp.down.weight",
76
+ "blocks.10.mlp.gate.weight",
77
+ "blocks.10.mlp.up.weight",
78
+ "blocks.11.attn.gate.weight",
79
+ "blocks.11.attn.wk.weight",
80
+ "blocks.11.attn.wq.weight",
81
+ "blocks.11.attn.wv.weight",
82
+ "blocks.5.mlp.down.weight",
83
+ "blocks.5.mlp.up.weight",
84
+ "blocks.6.attn.gate.weight",
85
+ "blocks.6.attn.wk.weight",
86
+ "blocks.6.attn.wo.weight",
87
+ "blocks.6.attn.wq.weight",
88
+ "blocks.6.attn.wv.weight",
89
+ "blocks.6.mlp.down.weight",
90
+ "blocks.6.mlp.gate.weight",
91
+ "blocks.6.mlp.up.weight",
92
+ "blocks.7.attn.gate.weight",
93
+ "blocks.7.attn.wk.weight",
94
+ "blocks.7.attn.wo.weight",
95
+ "blocks.7.attn.wq.weight",
96
+ "blocks.7.attn.wv.weight",
97
+ "blocks.7.mlp.down.weight",
98
+ "blocks.7.mlp.gate.weight",
99
+ "blocks.7.mlp.up.weight",
100
+ "blocks.8.attn.gate.weight",
101
+ "blocks.8.attn.wk.weight",
102
+ "blocks.8.attn.wo.weight",
103
+ "blocks.8.attn.wq.weight",
104
+ "blocks.8.attn.wv.weight",
105
+ "blocks.8.mlp.down.weight",
106
+ "blocks.8.mlp.gate.weight",
107
+ "blocks.8.mlp.up.weight",
108
+ "blocks.9.attn.gate.weight",
109
+ "blocks.9.attn.wk.weight",
110
+ "blocks.9.attn.wo.weight",
111
+ "blocks.9.attn.wq.weight",
112
+ "blocks.9.attn.wv.weight",
113
+ "blocks.9.mlp.down.weight",
114
+ "blocks.9.mlp.gate.weight",
115
+ "blocks.9.mlp.up.weight",
116
+ "blocks.11.attn.wo.weight",
117
+ "blocks.11.mlp.down.weight",
118
+ "blocks.11.mlp.gate.weight",
119
+ "blocks.11.mlp.up.weight",
120
+ "blocks.12.attn.gate.weight",
121
+ "blocks.12.attn.wk.weight",
122
+ "blocks.12.attn.wo.weight",
123
+ "blocks.12.attn.wq.weight",
124
+ "blocks.12.attn.wv.weight",
125
+ "blocks.12.mlp.down.weight",
126
+ "blocks.12.mlp.gate.weight",
127
+ "blocks.12.mlp.up.weight",
128
+ "blocks.13.attn.gate.weight",
129
+ "blocks.13.attn.wk.weight",
130
+ "blocks.13.attn.wo.weight",
131
+ "blocks.13.attn.wq.weight",
132
+ "blocks.13.attn.wv.weight",
133
+ "blocks.13.mlp.down.weight",
134
+ "blocks.13.mlp.gate.weight",
135
+ "blocks.13.mlp.up.weight",
136
+ "blocks.14.attn.gate.weight",
137
+ "blocks.14.attn.wk.weight",
138
+ "blocks.14.attn.wo.weight",
139
+ "blocks.14.attn.wq.weight",
140
+ "blocks.14.attn.wv.weight",
141
+ "blocks.14.mlp.down.weight",
142
+ "blocks.14.mlp.gate.weight",
143
+ "blocks.14.mlp.up.weight",
144
+ "blocks.15.attn.gate.weight",
145
+ "blocks.15.attn.wk.weight",
146
+ "blocks.15.attn.wo.weight",
147
+ "blocks.15.attn.wq.weight",
148
+ "blocks.15.attn.wv.weight",
149
+ "blocks.15.mlp.down.weight",
150
+ "blocks.15.mlp.gate.weight",
151
+ "blocks.15.mlp.up.weight",
152
+ "blocks.16.attn.gate.weight",
153
+ "blocks.16.attn.wk.weight",
154
+ "blocks.16.attn.wo.weight",
155
+ "blocks.16.attn.wq.weight",
156
+ "blocks.16.attn.wv.weight",
157
+ "blocks.16.mlp.gate.weight",
158
+ "blocks.16.mlp.up.weight",
159
+ "blocks.16.mlp.down.weight",
160
+ "blocks.17.attn.gate.weight",
161
+ "blocks.17.attn.wk.weight",
162
+ "blocks.17.attn.wo.weight",
163
+ "blocks.17.attn.wq.weight",
164
+ "blocks.17.attn.wv.weight",
165
+ "blocks.17.mlp.down.weight",
166
+ "blocks.17.mlp.gate.weight",
167
+ "blocks.17.mlp.up.weight",
168
+ "blocks.18.attn.gate.weight",
169
+ "blocks.18.attn.wk.weight",
170
+ "blocks.18.attn.wo.weight",
171
+ "blocks.18.attn.wq.weight",
172
+ "blocks.18.attn.wv.weight",
173
+ "blocks.18.mlp.down.weight",
174
+ "blocks.18.mlp.gate.weight",
175
+ "blocks.18.mlp.up.weight",
176
+ "blocks.19.attn.gate.weight",
177
+ "blocks.19.attn.wk.weight",
178
+ "blocks.19.attn.wo.weight",
179
+ "blocks.19.attn.wq.weight",
180
+ "blocks.19.attn.wv.weight",
181
+ "blocks.19.mlp.down.weight",
182
+ "blocks.19.mlp.gate.weight",
183
+ "blocks.19.mlp.up.weight",
184
+ "blocks.20.attn.gate.weight",
185
+ "blocks.20.attn.wk.weight",
186
+ "blocks.20.attn.wo.weight",
187
+ "blocks.20.attn.wq.weight",
188
+ "blocks.20.attn.wv.weight",
189
+ "blocks.20.mlp.down.weight",
190
+ "blocks.20.mlp.gate.weight",
191
+ "blocks.20.mlp.up.weight",
192
+ "blocks.21.attn.gate.weight",
193
+ "blocks.21.attn.wk.weight",
194
+ "blocks.21.attn.wo.weight",
195
+ "blocks.21.attn.wq.weight",
196
+ "blocks.21.attn.wv.weight",
197
+ "blocks.21.mlp.down.weight",
198
+ "blocks.21.mlp.gate.weight",
199
+ "blocks.21.mlp.up.weight",
200
+ "blocks.22.attn.gate.weight",
201
+ "blocks.22.attn.wk.weight",
202
+ "blocks.22.attn.wo.weight",
203
+ "blocks.22.attn.wq.weight",
204
+ "blocks.22.attn.wv.weight",
205
+ "blocks.22.mlp.down.weight",
206
+ "blocks.22.mlp.gate.weight",
207
+ "blocks.22.mlp.up.weight",
208
+ "blocks.23.attn.gate.weight",
209
+ "blocks.23.attn.wk.weight",
210
+ "blocks.23.attn.wo.weight",
211
+ "blocks.23.attn.wq.weight",
212
+ "blocks.23.attn.wv.weight",
213
+ "blocks.23.mlp.down.weight",
214
+ "blocks.23.mlp.gate.weight",
215
+ "blocks.23.mlp.up.weight",
216
+ "blocks.24.attn.gate.weight",
217
+ "blocks.24.attn.wk.weight",
218
+ "blocks.24.attn.wo.weight",
219
+ "blocks.24.attn.wq.weight",
220
+ "blocks.24.attn.wv.weight",
221
+ "blocks.24.mlp.down.weight",
222
+ "blocks.24.mlp.gate.weight",
223
+ "blocks.24.mlp.up.weight",
224
+ "blocks.25.attn.gate.weight",
225
+ "blocks.25.attn.wk.weight",
226
+ "blocks.25.attn.wo.weight",
227
+ "blocks.25.attn.wq.weight",
228
+ "blocks.25.attn.wv.weight",
229
+ "blocks.25.mlp.down.weight",
230
+ "blocks.25.mlp.gate.weight",
231
+ "blocks.25.mlp.up.weight",
232
+ "blocks.26.attn.gate.weight",
233
+ "blocks.26.attn.wk.weight",
234
+ "blocks.26.attn.wo.weight",
235
+ "blocks.26.attn.wq.weight",
236
+ "blocks.26.attn.wv.weight",
237
+ "blocks.26.mlp.down.weight",
238
+ "blocks.26.mlp.gate.weight",
239
+ "blocks.26.mlp.up.weight",
240
+ "blocks.27.attn.gate.weight",
241
+ "blocks.27.attn.wk.weight",
242
+ "blocks.27.attn.wo.weight",
243
+ "blocks.27.attn.wq.weight",
244
+ "blocks.27.attn.wv.weight",
245
+ "blocks.27.mlp.down.weight",
246
+ "blocks.27.mlp.gate.weight",
247
+ "blocks.27.mlp.up.weight",
248
+ "txtmlp.1.weight",
249
+ "txtmlp.3.weight",
250
+ "txtfusion.layerwise_blocks.0.attn.gate.weight",
251
+ "txtfusion.layerwise_blocks.0.attn.wk.weight",
252
+ "txtfusion.layerwise_blocks.0.attn.wo.weight",
253
+ "txtfusion.layerwise_blocks.0.attn.wq.weight",
254
+ "txtfusion.layerwise_blocks.0.attn.wv.weight",
255
+ "txtfusion.layerwise_blocks.0.mlp.down.weight",
256
+ "txtfusion.layerwise_blocks.0.mlp.gate.weight",
257
+ "txtfusion.layerwise_blocks.0.mlp.up.weight",
258
+ "txtfusion.layerwise_blocks.1.attn.gate.weight",
259
+ "txtfusion.layerwise_blocks.1.attn.wk.weight",
260
+ "txtfusion.layerwise_blocks.1.attn.wo.weight",
261
+ "txtfusion.layerwise_blocks.1.attn.wq.weight",
262
+ "txtfusion.layerwise_blocks.1.attn.wv.weight",
263
+ "txtfusion.layerwise_blocks.1.mlp.down.weight",
264
+ "txtfusion.layerwise_blocks.1.mlp.gate.weight",
265
+ "txtfusion.layerwise_blocks.1.mlp.up.weight",
266
+ "txtfusion.refiner_blocks.0.attn.gate.weight",
267
+ "txtfusion.refiner_blocks.0.attn.wk.weight",
268
+ "txtfusion.refiner_blocks.0.attn.wo.weight",
269
+ "txtfusion.refiner_blocks.0.attn.wq.weight",
270
+ "txtfusion.refiner_blocks.0.attn.wv.weight",
271
+ "txtfusion.refiner_blocks.0.mlp.down.weight",
272
+ "txtfusion.refiner_blocks.0.mlp.gate.weight",
273
+ "txtfusion.refiner_blocks.0.mlp.up.weight",
274
+ "txtfusion.refiner_blocks.1.attn.gate.weight",
275
+ "txtfusion.refiner_blocks.1.attn.wk.weight",
276
+ "txtfusion.refiner_blocks.1.attn.wo.weight",
277
+ "txtfusion.refiner_blocks.1.attn.wq.weight",
278
+ "txtfusion.refiner_blocks.1.attn.wv.weight",
279
+ "txtfusion.refiner_blocks.1.mlp.down.weight",
280
+ "txtfusion.refiner_blocks.1.mlp.gate.weight",
281
+ "txtfusion.refiner_blocks.1.mlp.up.weight"
282
+ ]
283
+ },
284
+ "modules_quant_config": {},
285
+ "modules_to_not_convert": [
286
+ ".txt_out",
287
+ ".time_embed",
288
+ ".vid_out",
289
+ ".vid_in",
290
+ ".txt_in",
291
+ ".img_in",
292
+ ".emb_out",
293
+ "first",
294
+ ".norm_out",
295
+ "multi_modal_projector",
296
+ ".img_out",
297
+ "tproj",
298
+ ".t_embedder",
299
+ ".emb_in",
300
+ ".final_layer",
301
+ "last",
302
+ "patch_embedding",
303
+ ".proj_out",
304
+ "time_text_embed",
305
+ "tmlp",
306
+ "patch_embed",
307
+ ".x_embedder",
308
+ "patch_emb",
309
+ "wte",
310
+ ".y_embedder",
311
+ ".condition_embedder",
312
+ ".context_embedder",
313
+ "projector",
314
+ "lm_head"
315
+ ],
316
+ "modules_to_not_use_matmul": [],
317
+ "non_blocking": false,
318
+ "quant_conv": false,
319
+ "quant_embedding": false,
320
+ "quant_method": "sdnq",
321
+ "quantization_device": "cuda",
322
+ "quantized_matmul_dtype": null,
323
+ "return_device": "cpu",
324
+ "sdnq_version": "0.2.1",
325
+ "svd_rank": 32,
326
+ "svd_steps": 8,
327
+ "use_dynamic_quantization": true,
328
+ "use_grad_ckpt": true,
329
+ "use_hadamard": true,
330
+ "use_quantized_matmul": true,
331
+ "use_quantized_matmul_conv": true,
332
+ "use_static_quantization": true,
333
+ "use_stochastic_rounding": false,
334
+ "use_svd": false,
335
+ "weights_dtype": "uint4"
336
+ },
337
+ "tdim": 256,
338
+ "theta": 1000.0,
339
+ "txtdim": 2560,
340
+ "txtheads": 20,
341
+ "txtkvheads": 20,
342
+ "txtlayers": 12
343
+ }
transformer/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ced334339fd4d2a696198014486b95ad6da849101fe5d1273072903324d7357
3
+ size 7747768840
transformer/quantization_config.json ADDED
@@ -0,0 +1,323 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_skip_keys": true,
3
+ "dequantize_fp32": false,
4
+ "dynamic_loss_threshold": -1,
5
+ "group_size": 0,
6
+ "hadamard_group_size": 256,
7
+ "is_integer": true,
8
+ "is_training": false,
9
+ "modules_dtype_dict": {
10
+ "uint4": [
11
+ "blocks.0.attn.gate.weight",
12
+ "blocks.0.attn.wk.weight",
13
+ "blocks.0.attn.wo.weight",
14
+ "blocks.0.attn.wq.weight",
15
+ "blocks.0.attn.wv.weight",
16
+ "blocks.0.mlp.down.weight",
17
+ "blocks.0.mlp.gate.weight",
18
+ "blocks.0.mlp.up.weight",
19
+ "blocks.1.attn.gate.weight",
20
+ "blocks.1.attn.wk.weight",
21
+ "blocks.1.attn.wo.weight",
22
+ "blocks.1.attn.wq.weight",
23
+ "blocks.1.attn.wv.weight",
24
+ "blocks.1.mlp.down.weight",
25
+ "blocks.1.mlp.gate.weight",
26
+ "blocks.1.mlp.up.weight",
27
+ "blocks.2.attn.gate.weight",
28
+ "blocks.2.attn.wk.weight",
29
+ "blocks.2.attn.wo.weight",
30
+ "blocks.2.attn.wq.weight",
31
+ "blocks.2.attn.wv.weight",
32
+ "blocks.2.mlp.down.weight",
33
+ "blocks.2.mlp.gate.weight",
34
+ "blocks.2.mlp.up.weight",
35
+ "blocks.3.attn.gate.weight",
36
+ "blocks.3.attn.wk.weight",
37
+ "blocks.3.attn.wo.weight",
38
+ "blocks.3.attn.wq.weight",
39
+ "blocks.3.attn.wv.weight",
40
+ "blocks.3.mlp.down.weight",
41
+ "blocks.3.mlp.gate.weight",
42
+ "blocks.3.mlp.up.weight",
43
+ "blocks.4.attn.gate.weight",
44
+ "blocks.4.attn.wk.weight",
45
+ "blocks.4.attn.wo.weight",
46
+ "blocks.4.attn.wq.weight",
47
+ "blocks.4.attn.wv.weight",
48
+ "blocks.4.mlp.down.weight",
49
+ "blocks.4.mlp.gate.weight",
50
+ "blocks.4.mlp.up.weight",
51
+ "blocks.5.attn.gate.weight",
52
+ "blocks.5.attn.wk.weight",
53
+ "blocks.5.attn.wo.weight",
54
+ "blocks.5.attn.wq.weight",
55
+ "blocks.5.attn.wv.weight",
56
+ "blocks.5.mlp.gate.weight",
57
+ "blocks.10.attn.gate.weight",
58
+ "blocks.10.attn.wk.weight",
59
+ "blocks.10.attn.wo.weight",
60
+ "blocks.10.attn.wq.weight",
61
+ "blocks.10.attn.wv.weight",
62
+ "blocks.10.mlp.down.weight",
63
+ "blocks.10.mlp.gate.weight",
64
+ "blocks.10.mlp.up.weight",
65
+ "blocks.11.attn.gate.weight",
66
+ "blocks.11.attn.wk.weight",
67
+ "blocks.11.attn.wq.weight",
68
+ "blocks.11.attn.wv.weight",
69
+ "blocks.5.mlp.down.weight",
70
+ "blocks.5.mlp.up.weight",
71
+ "blocks.6.attn.gate.weight",
72
+ "blocks.6.attn.wk.weight",
73
+ "blocks.6.attn.wo.weight",
74
+ "blocks.6.attn.wq.weight",
75
+ "blocks.6.attn.wv.weight",
76
+ "blocks.6.mlp.down.weight",
77
+ "blocks.6.mlp.gate.weight",
78
+ "blocks.6.mlp.up.weight",
79
+ "blocks.7.attn.gate.weight",
80
+ "blocks.7.attn.wk.weight",
81
+ "blocks.7.attn.wo.weight",
82
+ "blocks.7.attn.wq.weight",
83
+ "blocks.7.attn.wv.weight",
84
+ "blocks.7.mlp.down.weight",
85
+ "blocks.7.mlp.gate.weight",
86
+ "blocks.7.mlp.up.weight",
87
+ "blocks.8.attn.gate.weight",
88
+ "blocks.8.attn.wk.weight",
89
+ "blocks.8.attn.wo.weight",
90
+ "blocks.8.attn.wq.weight",
91
+ "blocks.8.attn.wv.weight",
92
+ "blocks.8.mlp.down.weight",
93
+ "blocks.8.mlp.gate.weight",
94
+ "blocks.8.mlp.up.weight",
95
+ "blocks.9.attn.gate.weight",
96
+ "blocks.9.attn.wk.weight",
97
+ "blocks.9.attn.wo.weight",
98
+ "blocks.9.attn.wq.weight",
99
+ "blocks.9.attn.wv.weight",
100
+ "blocks.9.mlp.down.weight",
101
+ "blocks.9.mlp.gate.weight",
102
+ "blocks.9.mlp.up.weight",
103
+ "blocks.11.attn.wo.weight",
104
+ "blocks.11.mlp.down.weight",
105
+ "blocks.11.mlp.gate.weight",
106
+ "blocks.11.mlp.up.weight",
107
+ "blocks.12.attn.gate.weight",
108
+ "blocks.12.attn.wk.weight",
109
+ "blocks.12.attn.wo.weight",
110
+ "blocks.12.attn.wq.weight",
111
+ "blocks.12.attn.wv.weight",
112
+ "blocks.12.mlp.down.weight",
113
+ "blocks.12.mlp.gate.weight",
114
+ "blocks.12.mlp.up.weight",
115
+ "blocks.13.attn.gate.weight",
116
+ "blocks.13.attn.wk.weight",
117
+ "blocks.13.attn.wo.weight",
118
+ "blocks.13.attn.wq.weight",
119
+ "blocks.13.attn.wv.weight",
120
+ "blocks.13.mlp.down.weight",
121
+ "blocks.13.mlp.gate.weight",
122
+ "blocks.13.mlp.up.weight",
123
+ "blocks.14.attn.gate.weight",
124
+ "blocks.14.attn.wk.weight",
125
+ "blocks.14.attn.wo.weight",
126
+ "blocks.14.attn.wq.weight",
127
+ "blocks.14.attn.wv.weight",
128
+ "blocks.14.mlp.down.weight",
129
+ "blocks.14.mlp.gate.weight",
130
+ "blocks.14.mlp.up.weight",
131
+ "blocks.15.attn.gate.weight",
132
+ "blocks.15.attn.wk.weight",
133
+ "blocks.15.attn.wo.weight",
134
+ "blocks.15.attn.wq.weight",
135
+ "blocks.15.attn.wv.weight",
136
+ "blocks.15.mlp.down.weight",
137
+ "blocks.15.mlp.gate.weight",
138
+ "blocks.15.mlp.up.weight",
139
+ "blocks.16.attn.gate.weight",
140
+ "blocks.16.attn.wk.weight",
141
+ "blocks.16.attn.wo.weight",
142
+ "blocks.16.attn.wq.weight",
143
+ "blocks.16.attn.wv.weight",
144
+ "blocks.16.mlp.gate.weight",
145
+ "blocks.16.mlp.up.weight",
146
+ "blocks.16.mlp.down.weight",
147
+ "blocks.17.attn.gate.weight",
148
+ "blocks.17.attn.wk.weight",
149
+ "blocks.17.attn.wo.weight",
150
+ "blocks.17.attn.wq.weight",
151
+ "blocks.17.attn.wv.weight",
152
+ "blocks.17.mlp.down.weight",
153
+ "blocks.17.mlp.gate.weight",
154
+ "blocks.17.mlp.up.weight",
155
+ "blocks.18.attn.gate.weight",
156
+ "blocks.18.attn.wk.weight",
157
+ "blocks.18.attn.wo.weight",
158
+ "blocks.18.attn.wq.weight",
159
+ "blocks.18.attn.wv.weight",
160
+ "blocks.18.mlp.down.weight",
161
+ "blocks.18.mlp.gate.weight",
162
+ "blocks.18.mlp.up.weight",
163
+ "blocks.19.attn.gate.weight",
164
+ "blocks.19.attn.wk.weight",
165
+ "blocks.19.attn.wo.weight",
166
+ "blocks.19.attn.wq.weight",
167
+ "blocks.19.attn.wv.weight",
168
+ "blocks.19.mlp.down.weight",
169
+ "blocks.19.mlp.gate.weight",
170
+ "blocks.19.mlp.up.weight",
171
+ "blocks.20.attn.gate.weight",
172
+ "blocks.20.attn.wk.weight",
173
+ "blocks.20.attn.wo.weight",
174
+ "blocks.20.attn.wq.weight",
175
+ "blocks.20.attn.wv.weight",
176
+ "blocks.20.mlp.down.weight",
177
+ "blocks.20.mlp.gate.weight",
178
+ "blocks.20.mlp.up.weight",
179
+ "blocks.21.attn.gate.weight",
180
+ "blocks.21.attn.wk.weight",
181
+ "blocks.21.attn.wo.weight",
182
+ "blocks.21.attn.wq.weight",
183
+ "blocks.21.attn.wv.weight",
184
+ "blocks.21.mlp.down.weight",
185
+ "blocks.21.mlp.gate.weight",
186
+ "blocks.21.mlp.up.weight",
187
+ "blocks.22.attn.gate.weight",
188
+ "blocks.22.attn.wk.weight",
189
+ "blocks.22.attn.wo.weight",
190
+ "blocks.22.attn.wq.weight",
191
+ "blocks.22.attn.wv.weight",
192
+ "blocks.22.mlp.down.weight",
193
+ "blocks.22.mlp.gate.weight",
194
+ "blocks.22.mlp.up.weight",
195
+ "blocks.23.attn.gate.weight",
196
+ "blocks.23.attn.wk.weight",
197
+ "blocks.23.attn.wo.weight",
198
+ "blocks.23.attn.wq.weight",
199
+ "blocks.23.attn.wv.weight",
200
+ "blocks.23.mlp.down.weight",
201
+ "blocks.23.mlp.gate.weight",
202
+ "blocks.23.mlp.up.weight",
203
+ "blocks.24.attn.gate.weight",
204
+ "blocks.24.attn.wk.weight",
205
+ "blocks.24.attn.wo.weight",
206
+ "blocks.24.attn.wq.weight",
207
+ "blocks.24.attn.wv.weight",
208
+ "blocks.24.mlp.down.weight",
209
+ "blocks.24.mlp.gate.weight",
210
+ "blocks.24.mlp.up.weight",
211
+ "blocks.25.attn.gate.weight",
212
+ "blocks.25.attn.wk.weight",
213
+ "blocks.25.attn.wo.weight",
214
+ "blocks.25.attn.wq.weight",
215
+ "blocks.25.attn.wv.weight",
216
+ "blocks.25.mlp.down.weight",
217
+ "blocks.25.mlp.gate.weight",
218
+ "blocks.25.mlp.up.weight",
219
+ "blocks.26.attn.gate.weight",
220
+ "blocks.26.attn.wk.weight",
221
+ "blocks.26.attn.wo.weight",
222
+ "blocks.26.attn.wq.weight",
223
+ "blocks.26.attn.wv.weight",
224
+ "blocks.26.mlp.down.weight",
225
+ "blocks.26.mlp.gate.weight",
226
+ "blocks.26.mlp.up.weight",
227
+ "blocks.27.attn.gate.weight",
228
+ "blocks.27.attn.wk.weight",
229
+ "blocks.27.attn.wo.weight",
230
+ "blocks.27.attn.wq.weight",
231
+ "blocks.27.attn.wv.weight",
232
+ "blocks.27.mlp.down.weight",
233
+ "blocks.27.mlp.gate.weight",
234
+ "blocks.27.mlp.up.weight",
235
+ "txtmlp.1.weight",
236
+ "txtmlp.3.weight",
237
+ "txtfusion.layerwise_blocks.0.attn.gate.weight",
238
+ "txtfusion.layerwise_blocks.0.attn.wk.weight",
239
+ "txtfusion.layerwise_blocks.0.attn.wo.weight",
240
+ "txtfusion.layerwise_blocks.0.attn.wq.weight",
241
+ "txtfusion.layerwise_blocks.0.attn.wv.weight",
242
+ "txtfusion.layerwise_blocks.0.mlp.down.weight",
243
+ "txtfusion.layerwise_blocks.0.mlp.gate.weight",
244
+ "txtfusion.layerwise_blocks.0.mlp.up.weight",
245
+ "txtfusion.layerwise_blocks.1.attn.gate.weight",
246
+ "txtfusion.layerwise_blocks.1.attn.wk.weight",
247
+ "txtfusion.layerwise_blocks.1.attn.wo.weight",
248
+ "txtfusion.layerwise_blocks.1.attn.wq.weight",
249
+ "txtfusion.layerwise_blocks.1.attn.wv.weight",
250
+ "txtfusion.layerwise_blocks.1.mlp.down.weight",
251
+ "txtfusion.layerwise_blocks.1.mlp.gate.weight",
252
+ "txtfusion.layerwise_blocks.1.mlp.up.weight",
253
+ "txtfusion.refiner_blocks.0.attn.gate.weight",
254
+ "txtfusion.refiner_blocks.0.attn.wk.weight",
255
+ "txtfusion.refiner_blocks.0.attn.wo.weight",
256
+ "txtfusion.refiner_blocks.0.attn.wq.weight",
257
+ "txtfusion.refiner_blocks.0.attn.wv.weight",
258
+ "txtfusion.refiner_blocks.0.mlp.down.weight",
259
+ "txtfusion.refiner_blocks.0.mlp.gate.weight",
260
+ "txtfusion.refiner_blocks.0.mlp.up.weight",
261
+ "txtfusion.refiner_blocks.1.attn.gate.weight",
262
+ "txtfusion.refiner_blocks.1.attn.wk.weight",
263
+ "txtfusion.refiner_blocks.1.attn.wo.weight",
264
+ "txtfusion.refiner_blocks.1.attn.wq.weight",
265
+ "txtfusion.refiner_blocks.1.attn.wv.weight",
266
+ "txtfusion.refiner_blocks.1.mlp.down.weight",
267
+ "txtfusion.refiner_blocks.1.mlp.gate.weight",
268
+ "txtfusion.refiner_blocks.1.mlp.up.weight"
269
+ ]
270
+ },
271
+ "modules_quant_config": {},
272
+ "modules_to_not_convert": [
273
+ ".txt_out",
274
+ ".time_embed",
275
+ ".vid_out",
276
+ ".vid_in",
277
+ ".txt_in",
278
+ ".img_in",
279
+ ".emb_out",
280
+ "first",
281
+ ".norm_out",
282
+ "multi_modal_projector",
283
+ ".img_out",
284
+ "tproj",
285
+ ".t_embedder",
286
+ ".emb_in",
287
+ ".final_layer",
288
+ "last",
289
+ "patch_embedding",
290
+ ".proj_out",
291
+ "time_text_embed",
292
+ "tmlp",
293
+ "patch_embed",
294
+ ".x_embedder",
295
+ "patch_emb",
296
+ "wte",
297
+ ".y_embedder",
298
+ ".condition_embedder",
299
+ ".context_embedder",
300
+ "projector",
301
+ "lm_head"
302
+ ],
303
+ "modules_to_not_use_matmul": [],
304
+ "non_blocking": false,
305
+ "quant_conv": false,
306
+ "quant_embedding": false,
307
+ "quant_method": "sdnq",
308
+ "quantization_device": "cuda",
309
+ "quantized_matmul_dtype": null,
310
+ "return_device": "cpu",
311
+ "sdnq_version": "0.2.1",
312
+ "svd_rank": 32,
313
+ "svd_steps": 8,
314
+ "use_dynamic_quantization": true,
315
+ "use_grad_ckpt": true,
316
+ "use_hadamard": true,
317
+ "use_quantized_matmul": true,
318
+ "use_quantized_matmul_conv": true,
319
+ "use_static_quantization": true,
320
+ "use_stochastic_rounding": false,
321
+ "use_svd": false,
322
+ "weights_dtype": "uint4"
323
+ }
vae/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKLQwenImage",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "/mnt/models/Diffusers/models--CalamitousFelicitousness--Krea-2-Base-Diffusers/snapshots/242c829a59107d9f5970216acfad6d2e9000e750/vae",
5
+ "attn_scales": [],
6
+ "base_dim": 96,
7
+ "dim_mult": [
8
+ 1,
9
+ 2,
10
+ 4,
11
+ 4
12
+ ],
13
+ "dropout": 0.0,
14
+ "input_channels": 3,
15
+ "latents_mean": [
16
+ -0.7571,
17
+ -0.7089,
18
+ -0.9113,
19
+ 0.1075,
20
+ -0.1745,
21
+ 0.9653,
22
+ -0.1517,
23
+ 1.5508,
24
+ 0.4134,
25
+ -0.0715,
26
+ 0.5517,
27
+ -0.3632,
28
+ -0.1922,
29
+ -0.9497,
30
+ 0.2503,
31
+ -0.2921
32
+ ],
33
+ "latents_std": [
34
+ 2.8184,
35
+ 1.4541,
36
+ 2.3275,
37
+ 2.6558,
38
+ 1.2196,
39
+ 1.7708,
40
+ 2.6052,
41
+ 2.0743,
42
+ 3.2687,
43
+ 2.1526,
44
+ 2.8652,
45
+ 1.5579,
46
+ 1.6382,
47
+ 1.1253,
48
+ 2.8251,
49
+ 1.916
50
+ ],
51
+ "num_res_blocks": 2,
52
+ "temperal_downsample": [
53
+ false,
54
+ true,
55
+ true
56
+ ],
57
+ "z_dim": 16
58
+ }
vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c8bc8b758c649abef9ea407b95408389a3b2f610d0d10fcb054fe171d0a8344
3
+ size 253806966