codeShare commited on
Commit
2390171
·
verified ·
1 Parent(s): 4beb777

Direct upload of rebuilt_pipeline_temp contents

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
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.38.0",
4
+ "base_image_seq_len": 256,
5
+ "base_shift": 0.5,
6
+ "invert_sigmas": false,
7
+ "max_image_seq_len": 4096,
8
+ "max_shift": 1.15,
9
+ "num_train_timesteps": 1000,
10
+ "shift": 3.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,384 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 151645,
10
+ "head_dim": 128,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 4096,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 12288,
15
+ "layer_types": [
16
+ "full_attention",
17
+ "full_attention",
18
+ "full_attention",
19
+ "full_attention",
20
+ "full_attention",
21
+ "full_attention",
22
+ "full_attention",
23
+ "full_attention",
24
+ "full_attention",
25
+ "full_attention",
26
+ "full_attention",
27
+ "full_attention",
28
+ "full_attention",
29
+ "full_attention",
30
+ "full_attention",
31
+ "full_attention",
32
+ "full_attention",
33
+ "full_attention",
34
+ "full_attention",
35
+ "full_attention",
36
+ "full_attention",
37
+ "full_attention",
38
+ "full_attention",
39
+ "full_attention",
40
+ "full_attention",
41
+ "full_attention",
42
+ "full_attention",
43
+ "full_attention",
44
+ "full_attention",
45
+ "full_attention",
46
+ "full_attention",
47
+ "full_attention",
48
+ "full_attention",
49
+ "full_attention",
50
+ "full_attention",
51
+ "full_attention"
52
+ ],
53
+ "max_position_embeddings": 40960,
54
+ "max_window_layers": 36,
55
+ "model_type": "qwen3",
56
+ "num_attention_heads": 32,
57
+ "num_hidden_layers": 36,
58
+ "num_key_value_heads": 8,
59
+ "pad_token_id": null,
60
+ "quantization_config": {
61
+ "add_skip_keys": true,
62
+ "dequantize_fp32": true,
63
+ "dynamic_loss_threshold": 0.01,
64
+ "group_size": 0,
65
+ "is_integer": true,
66
+ "is_training": false,
67
+ "modules_dtype_dict": {
68
+ "int5": [
69
+ "model.layers.1.mlp.down_proj.weight",
70
+ "model.layers.2.mlp.down_proj.weight",
71
+ "model.layers.3.mlp.down_proj.weight",
72
+ "model.layers.24.self_attn.v_proj.weight"
73
+ ],
74
+ "uint4": [
75
+ "model.layers.0.self_attn.q_proj.weight",
76
+ "model.layers.0.self_attn.k_proj.weight",
77
+ "model.layers.0.self_attn.v_proj.weight",
78
+ "model.layers.0.self_attn.o_proj.weight",
79
+ "model.layers.0.mlp.gate_proj.weight",
80
+ "model.layers.0.mlp.up_proj.weight",
81
+ "model.layers.0.mlp.down_proj.weight",
82
+ "model.layers.1.self_attn.q_proj.weight",
83
+ "model.layers.1.self_attn.k_proj.weight",
84
+ "model.layers.1.self_attn.v_proj.weight",
85
+ "model.layers.1.self_attn.o_proj.weight",
86
+ "model.layers.1.mlp.gate_proj.weight",
87
+ "model.layers.1.mlp.up_proj.weight",
88
+ "model.layers.2.self_attn.q_proj.weight",
89
+ "model.layers.2.self_attn.k_proj.weight",
90
+ "model.layers.2.self_attn.v_proj.weight",
91
+ "model.layers.2.self_attn.o_proj.weight",
92
+ "model.layers.2.mlp.gate_proj.weight",
93
+ "model.layers.2.mlp.up_proj.weight",
94
+ "model.layers.3.self_attn.q_proj.weight",
95
+ "model.layers.3.self_attn.k_proj.weight",
96
+ "model.layers.3.self_attn.v_proj.weight",
97
+ "model.layers.3.self_attn.o_proj.weight",
98
+ "model.layers.3.mlp.gate_proj.weight",
99
+ "model.layers.3.mlp.up_proj.weight",
100
+ "model.layers.4.self_attn.q_proj.weight",
101
+ "model.layers.4.self_attn.k_proj.weight",
102
+ "model.layers.4.self_attn.v_proj.weight",
103
+ "model.layers.4.self_attn.o_proj.weight",
104
+ "model.layers.4.mlp.gate_proj.weight",
105
+ "model.layers.4.mlp.up_proj.weight",
106
+ "model.layers.4.mlp.down_proj.weight",
107
+ "model.layers.5.self_attn.q_proj.weight",
108
+ "model.layers.5.self_attn.k_proj.weight",
109
+ "model.layers.5.self_attn.v_proj.weight",
110
+ "model.layers.5.self_attn.o_proj.weight",
111
+ "model.layers.5.mlp.gate_proj.weight",
112
+ "model.layers.5.mlp.up_proj.weight",
113
+ "model.layers.5.mlp.down_proj.weight",
114
+ "model.layers.6.self_attn.q_proj.weight",
115
+ "model.layers.6.self_attn.k_proj.weight",
116
+ "model.layers.6.self_attn.v_proj.weight",
117
+ "model.layers.6.self_attn.o_proj.weight",
118
+ "model.layers.6.mlp.gate_proj.weight",
119
+ "model.layers.6.mlp.up_proj.weight",
120
+ "model.layers.6.mlp.down_proj.weight",
121
+ "model.layers.7.self_attn.q_proj.weight",
122
+ "model.layers.7.self_attn.k_proj.weight",
123
+ "model.layers.7.self_attn.v_proj.weight",
124
+ "model.layers.7.self_attn.o_proj.weight",
125
+ "model.layers.7.mlp.gate_proj.weight",
126
+ "model.layers.7.mlp.up_proj.weight",
127
+ "model.layers.7.mlp.down_proj.weight",
128
+ "model.layers.8.self_attn.q_proj.weight",
129
+ "model.layers.8.self_attn.k_proj.weight",
130
+ "model.layers.8.self_attn.v_proj.weight",
131
+ "model.layers.8.self_attn.o_proj.weight",
132
+ "model.layers.8.mlp.gate_proj.weight",
133
+ "model.layers.8.mlp.up_proj.weight",
134
+ "model.layers.8.mlp.down_proj.weight",
135
+ "model.layers.9.self_attn.q_proj.weight",
136
+ "model.layers.9.self_attn.k_proj.weight",
137
+ "model.layers.9.self_attn.v_proj.weight",
138
+ "model.layers.9.self_attn.o_proj.weight",
139
+ "model.layers.9.mlp.gate_proj.weight",
140
+ "model.layers.9.mlp.up_proj.weight",
141
+ "model.layers.9.mlp.down_proj.weight",
142
+ "model.layers.10.self_attn.q_proj.weight",
143
+ "model.layers.10.self_attn.k_proj.weight",
144
+ "model.layers.10.self_attn.v_proj.weight",
145
+ "model.layers.10.self_attn.o_proj.weight",
146
+ "model.layers.10.mlp.gate_proj.weight",
147
+ "model.layers.10.mlp.up_proj.weight",
148
+ "model.layers.10.mlp.down_proj.weight",
149
+ "model.layers.11.self_attn.q_proj.weight",
150
+ "model.layers.11.self_attn.k_proj.weight",
151
+ "model.layers.11.self_attn.v_proj.weight",
152
+ "model.layers.11.self_attn.o_proj.weight",
153
+ "model.layers.11.mlp.gate_proj.weight",
154
+ "model.layers.11.mlp.up_proj.weight",
155
+ "model.layers.11.mlp.down_proj.weight",
156
+ "model.layers.12.self_attn.q_proj.weight",
157
+ "model.layers.12.self_attn.k_proj.weight",
158
+ "model.layers.12.self_attn.v_proj.weight",
159
+ "model.layers.12.self_attn.o_proj.weight",
160
+ "model.layers.12.mlp.gate_proj.weight",
161
+ "model.layers.12.mlp.up_proj.weight",
162
+ "model.layers.12.mlp.down_proj.weight",
163
+ "model.layers.13.self_attn.q_proj.weight",
164
+ "model.layers.13.self_attn.k_proj.weight",
165
+ "model.layers.13.self_attn.v_proj.weight",
166
+ "model.layers.13.self_attn.o_proj.weight",
167
+ "model.layers.13.mlp.gate_proj.weight",
168
+ "model.layers.13.mlp.up_proj.weight",
169
+ "model.layers.13.mlp.down_proj.weight",
170
+ "model.layers.14.self_attn.q_proj.weight",
171
+ "model.layers.14.self_attn.k_proj.weight",
172
+ "model.layers.14.self_attn.v_proj.weight",
173
+ "model.layers.14.self_attn.o_proj.weight",
174
+ "model.layers.14.mlp.gate_proj.weight",
175
+ "model.layers.14.mlp.up_proj.weight",
176
+ "model.layers.14.mlp.down_proj.weight",
177
+ "model.layers.15.self_attn.q_proj.weight",
178
+ "model.layers.15.self_attn.k_proj.weight",
179
+ "model.layers.15.self_attn.v_proj.weight",
180
+ "model.layers.15.self_attn.o_proj.weight",
181
+ "model.layers.15.mlp.gate_proj.weight",
182
+ "model.layers.15.mlp.up_proj.weight",
183
+ "model.layers.15.mlp.down_proj.weight",
184
+ "model.layers.16.self_attn.q_proj.weight",
185
+ "model.layers.16.self_attn.k_proj.weight",
186
+ "model.layers.16.self_attn.v_proj.weight",
187
+ "model.layers.16.self_attn.o_proj.weight",
188
+ "model.layers.16.mlp.gate_proj.weight",
189
+ "model.layers.16.mlp.up_proj.weight",
190
+ "model.layers.16.mlp.down_proj.weight",
191
+ "model.layers.17.self_attn.q_proj.weight",
192
+ "model.layers.17.self_attn.k_proj.weight",
193
+ "model.layers.17.self_attn.v_proj.weight",
194
+ "model.layers.17.self_attn.o_proj.weight",
195
+ "model.layers.17.mlp.gate_proj.weight",
196
+ "model.layers.17.mlp.up_proj.weight",
197
+ "model.layers.17.mlp.down_proj.weight",
198
+ "model.layers.18.self_attn.q_proj.weight",
199
+ "model.layers.18.self_attn.k_proj.weight",
200
+ "model.layers.18.self_attn.v_proj.weight",
201
+ "model.layers.18.self_attn.o_proj.weight",
202
+ "model.layers.18.mlp.gate_proj.weight",
203
+ "model.layers.18.mlp.up_proj.weight",
204
+ "model.layers.18.mlp.down_proj.weight",
205
+ "model.layers.19.self_attn.q_proj.weight",
206
+ "model.layers.19.self_attn.k_proj.weight",
207
+ "model.layers.19.self_attn.v_proj.weight",
208
+ "model.layers.19.self_attn.o_proj.weight",
209
+ "model.layers.19.mlp.gate_proj.weight",
210
+ "model.layers.19.mlp.up_proj.weight",
211
+ "model.layers.19.mlp.down_proj.weight",
212
+ "model.layers.20.self_attn.q_proj.weight",
213
+ "model.layers.20.self_attn.k_proj.weight",
214
+ "model.layers.20.self_attn.v_proj.weight",
215
+ "model.layers.20.self_attn.o_proj.weight",
216
+ "model.layers.20.mlp.gate_proj.weight",
217
+ "model.layers.20.mlp.up_proj.weight",
218
+ "model.layers.20.mlp.down_proj.weight",
219
+ "model.layers.21.self_attn.q_proj.weight",
220
+ "model.layers.21.self_attn.k_proj.weight",
221
+ "model.layers.21.self_attn.v_proj.weight",
222
+ "model.layers.21.self_attn.o_proj.weight",
223
+ "model.layers.21.mlp.gate_proj.weight",
224
+ "model.layers.21.mlp.up_proj.weight",
225
+ "model.layers.21.mlp.down_proj.weight",
226
+ "model.layers.22.self_attn.q_proj.weight",
227
+ "model.layers.22.self_attn.k_proj.weight",
228
+ "model.layers.22.self_attn.v_proj.weight",
229
+ "model.layers.22.self_attn.o_proj.weight",
230
+ "model.layers.22.mlp.gate_proj.weight",
231
+ "model.layers.22.mlp.up_proj.weight",
232
+ "model.layers.22.mlp.down_proj.weight",
233
+ "model.layers.23.self_attn.q_proj.weight",
234
+ "model.layers.23.self_attn.k_proj.weight",
235
+ "model.layers.23.self_attn.v_proj.weight",
236
+ "model.layers.23.self_attn.o_proj.weight",
237
+ "model.layers.23.mlp.gate_proj.weight",
238
+ "model.layers.23.mlp.up_proj.weight",
239
+ "model.layers.23.mlp.down_proj.weight",
240
+ "model.layers.24.self_attn.q_proj.weight",
241
+ "model.layers.24.self_attn.k_proj.weight",
242
+ "model.layers.24.self_attn.o_proj.weight",
243
+ "model.layers.24.mlp.gate_proj.weight",
244
+ "model.layers.24.mlp.up_proj.weight",
245
+ "model.layers.24.mlp.down_proj.weight",
246
+ "model.layers.25.self_attn.q_proj.weight",
247
+ "model.layers.25.self_attn.k_proj.weight",
248
+ "model.layers.25.self_attn.v_proj.weight",
249
+ "model.layers.25.self_attn.o_proj.weight",
250
+ "model.layers.25.mlp.gate_proj.weight",
251
+ "model.layers.25.mlp.up_proj.weight",
252
+ "model.layers.25.mlp.down_proj.weight",
253
+ "model.layers.26.self_attn.q_proj.weight",
254
+ "model.layers.26.self_attn.k_proj.weight",
255
+ "model.layers.26.self_attn.v_proj.weight",
256
+ "model.layers.26.self_attn.o_proj.weight",
257
+ "model.layers.26.mlp.gate_proj.weight",
258
+ "model.layers.26.mlp.up_proj.weight",
259
+ "model.layers.26.mlp.down_proj.weight",
260
+ "model.layers.27.self_attn.q_proj.weight",
261
+ "model.layers.27.self_attn.k_proj.weight",
262
+ "model.layers.27.self_attn.v_proj.weight",
263
+ "model.layers.27.self_attn.o_proj.weight",
264
+ "model.layers.27.mlp.gate_proj.weight",
265
+ "model.layers.27.mlp.up_proj.weight",
266
+ "model.layers.27.mlp.down_proj.weight",
267
+ "model.layers.28.self_attn.q_proj.weight",
268
+ "model.layers.28.self_attn.k_proj.weight",
269
+ "model.layers.28.self_attn.v_proj.weight",
270
+ "model.layers.28.self_attn.o_proj.weight",
271
+ "model.layers.28.mlp.gate_proj.weight",
272
+ "model.layers.28.mlp.up_proj.weight",
273
+ "model.layers.28.mlp.down_proj.weight",
274
+ "model.layers.29.self_attn.q_proj.weight",
275
+ "model.layers.29.self_attn.k_proj.weight",
276
+ "model.layers.29.self_attn.v_proj.weight",
277
+ "model.layers.29.self_attn.o_proj.weight",
278
+ "model.layers.29.mlp.gate_proj.weight",
279
+ "model.layers.29.mlp.up_proj.weight",
280
+ "model.layers.29.mlp.down_proj.weight",
281
+ "model.layers.30.self_attn.q_proj.weight",
282
+ "model.layers.30.self_attn.k_proj.weight",
283
+ "model.layers.30.self_attn.v_proj.weight",
284
+ "model.layers.30.self_attn.o_proj.weight",
285
+ "model.layers.30.mlp.gate_proj.weight",
286
+ "model.layers.30.mlp.up_proj.weight",
287
+ "model.layers.30.mlp.down_proj.weight",
288
+ "model.layers.31.self_attn.q_proj.weight",
289
+ "model.layers.31.self_attn.k_proj.weight",
290
+ "model.layers.31.self_attn.v_proj.weight",
291
+ "model.layers.31.self_attn.o_proj.weight",
292
+ "model.layers.31.mlp.gate_proj.weight",
293
+ "model.layers.31.mlp.up_proj.weight",
294
+ "model.layers.31.mlp.down_proj.weight",
295
+ "model.layers.32.self_attn.q_proj.weight",
296
+ "model.layers.32.self_attn.k_proj.weight",
297
+ "model.layers.32.self_attn.v_proj.weight",
298
+ "model.layers.32.self_attn.o_proj.weight",
299
+ "model.layers.32.mlp.gate_proj.weight",
300
+ "model.layers.32.mlp.up_proj.weight",
301
+ "model.layers.32.mlp.down_proj.weight",
302
+ "model.layers.33.self_attn.q_proj.weight",
303
+ "model.layers.33.self_attn.k_proj.weight",
304
+ "model.layers.33.self_attn.v_proj.weight",
305
+ "model.layers.33.self_attn.o_proj.weight",
306
+ "model.layers.33.mlp.gate_proj.weight",
307
+ "model.layers.33.mlp.up_proj.weight",
308
+ "model.layers.33.mlp.down_proj.weight",
309
+ "model.layers.34.self_attn.q_proj.weight",
310
+ "model.layers.34.self_attn.k_proj.weight",
311
+ "model.layers.34.self_attn.v_proj.weight",
312
+ "model.layers.34.self_attn.o_proj.weight",
313
+ "model.layers.34.mlp.gate_proj.weight",
314
+ "model.layers.34.mlp.up_proj.weight",
315
+ "model.layers.34.mlp.down_proj.weight",
316
+ "model.layers.35.self_attn.q_proj.weight",
317
+ "model.layers.35.self_attn.k_proj.weight",
318
+ "model.layers.35.self_attn.v_proj.weight",
319
+ "model.layers.35.self_attn.o_proj.weight",
320
+ "model.layers.35.mlp.gate_proj.weight",
321
+ "model.layers.35.mlp.up_proj.weight",
322
+ "model.layers.35.mlp.down_proj.weight"
323
+ ]
324
+ },
325
+ "modules_quant_config": {},
326
+ "modules_to_not_convert": [
327
+ "patch_embedding",
328
+ ".y_embedder",
329
+ "model.embed_tokens.weight",
330
+ ".txt_in",
331
+ "time_text_embed",
332
+ "lm_head",
333
+ "multi_modal_projector",
334
+ ".condition_embedder",
335
+ ".img_in",
336
+ ".emb_out",
337
+ "wte",
338
+ ".final_layer",
339
+ ".t_embedder",
340
+ ".norm_out",
341
+ "patch_emb",
342
+ ".vid_in",
343
+ ".context_embedder",
344
+ ".vid_out",
345
+ "lm_head.weight",
346
+ ".x_embedder",
347
+ ".proj_out",
348
+ ".img_out",
349
+ ".time_embed",
350
+ ".emb_in",
351
+ ".txt_out",
352
+ "patch_embed"
353
+ ],
354
+ "non_blocking": false,
355
+ "quant_conv": false,
356
+ "quant_embedding": false,
357
+ "quant_method": "sdnq",
358
+ "quantization_device": "cpu",
359
+ "quantized_matmul_dtype": null,
360
+ "return_device": "cpu",
361
+ "sdnq_version": "0.1.8",
362
+ "svd_rank": 32,
363
+ "svd_steps": 8,
364
+ "use_dynamic_quantization": true,
365
+ "use_grad_ckpt": true,
366
+ "use_quantized_matmul": false,
367
+ "use_quantized_matmul_conv": false,
368
+ "use_static_quantization": true,
369
+ "use_stochastic_rounding": false,
370
+ "use_svd": false,
371
+ "weights_dtype": "uint4"
372
+ },
373
+ "rms_norm_eps": 1e-06,
374
+ "rope_parameters": {
375
+ "rope_theta": 1000000,
376
+ "rope_type": "default"
377
+ },
378
+ "sliding_window": null,
379
+ "tie_word_embeddings": false,
380
+ "transformers_version": "5.8.0",
381
+ "use_cache": true,
382
+ "use_sliding_window": false,
383
+ "vocab_size": 151936
384
+ }
text_encoder/generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 151643,
4
+ "eos_token_id": 151645,
5
+ "transformers_version": "5.8.0",
6
+ "use_cache": true
7
+ }
text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14dbce6fb2cc75c2d0ca2fb17e954c1248f3ebb8652b9d88d632aeb87f8a3a5d
3
+ size 6835992696
tokenizer/chat_template.jinja ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {{- messages[0].content + '\n\n' }}
5
+ {%- endif %}
6
+ {{- "# 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>" }}
7
+ {%- for tool in tools %}
8
+ {{- "\n" }}
9
+ {{- tool | tojson }}
10
+ {%- endfor %}
11
+ {{- "\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" }}
12
+ {%- else %}
13
+ {%- if messages[0].role == 'system' %}
14
+ {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
15
+ {%- endif %}
16
+ {%- endif %}
17
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
18
+ {%- for message in messages[::-1] %}
19
+ {%- set index = (messages|length - 1) - loop.index0 %}
20
+ {%- if ns.multi_step_tool and message.role == "user" and message.content is string and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
21
+ {%- set ns.multi_step_tool = false %}
22
+ {%- set ns.last_query_index = index %}
23
+ {%- endif %}
24
+ {%- endfor %}
25
+ {%- for message in messages %}
26
+ {%- if message.content is string %}
27
+ {%- set content = message.content %}
28
+ {%- else %}
29
+ {%- set content = '' %}
30
+ {%- endif %}
31
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
32
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
33
+ {%- elif message.role == "assistant" %}
34
+ {%- set reasoning_content = '' %}
35
+ {%- if message.reasoning_content is string %}
36
+ {%- set reasoning_content = message.reasoning_content %}
37
+ {%- else %}
38
+ {%- if '</think>' in content %}
39
+ {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
40
+ {%- set content = content.split('</think>')[-1].lstrip('\n') %}
41
+ {%- endif %}
42
+ {%- endif %}
43
+ {%- if loop.index0 > ns.last_query_index %}
44
+ {%- if loop.last or (not loop.last and reasoning_content) %}
45
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
46
+ {%- else %}
47
+ {{- '<|im_start|>' + message.role + '\n' + content }}
48
+ {%- endif %}
49
+ {%- else %}
50
+ {{- '<|im_start|>' + message.role + '\n' + content }}
51
+ {%- endif %}
52
+ {%- if message.tool_calls %}
53
+ {%- for tool_call in message.tool_calls %}
54
+ {%- if (loop.first and content) or (not loop.first) %}
55
+ {{- '\n' }}
56
+ {%- endif %}
57
+ {%- if tool_call.function %}
58
+ {%- set tool_call = tool_call.function %}
59
+ {%- endif %}
60
+ {{- '<tool_call>\n{"name": "' }}
61
+ {{- tool_call.name }}
62
+ {{- '", "arguments": ' }}
63
+ {%- if tool_call.arguments is string %}
64
+ {{- tool_call.arguments }}
65
+ {%- else %}
66
+ {{- tool_call.arguments | tojson }}
67
+ {%- endif %}
68
+ {{- '}\n</tool_call>' }}
69
+ {%- endfor %}
70
+ {%- endif %}
71
+ {{- '<|im_end|>\n' }}
72
+ {%- elif message.role == "tool" %}
73
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
74
+ {{- '<|im_start|>user' }}
75
+ {%- endif %}
76
+ {{- '\n<tool_response>\n' }}
77
+ {{- content }}
78
+ {{- '\n</tool_response>' }}
79
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
80
+ {{- '<|im_end|>\n' }}
81
+ {%- endif %}
82
+ {%- endif %}
83
+ {%- endfor %}
84
+ {%- if add_generation_prompt %}
85
+ {{- '<|im_start|>assistant\n' }}
86
+ {%- if enable_thinking is defined and enable_thinking is false %}
87
+ {{- '<think>\n\n</think>\n\n' }}
88
+ {%- endif %}
89
+ {%- 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,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "is_local": true,
9
+ "local_files_only": false,
10
+ "model_max_length": 131072,
11
+ "pad_token": "<|endoftext|>",
12
+ "split_special_tokens": false,
13
+ "tokenizer_class": "Qwen2Tokenizer",
14
+ "unk_token": null
15
+ }
transformer/config.json ADDED
@@ -0,0 +1,211 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Flux2Transformer2DModel",
3
+ "_diffusers_version": "0.38.0",
4
+ "_name_or_path": "/root/.cache/huggingface/hub/models--black-forest-labs--FLUX.2-klein-9B/snapshots/92196c8e11f7b6cf2b7493e037d8c5345c559216/transformer",
5
+ "attention_head_dim": 128,
6
+ "axes_dims_rope": [
7
+ 32,
8
+ 32,
9
+ 32,
10
+ 32
11
+ ],
12
+ "eps": 1e-06,
13
+ "guidance_embeds": false,
14
+ "in_channels": 128,
15
+ "joint_attention_dim": 12288,
16
+ "mlp_ratio": 3.0,
17
+ "num_attention_heads": 32,
18
+ "num_layers": 8,
19
+ "num_single_layers": 24,
20
+ "out_channels": null,
21
+ "patch_size": 1,
22
+ "quantization_config": {
23
+ "add_skip_keys": true,
24
+ "dequantize_fp32": true,
25
+ "dynamic_loss_threshold": 0.01,
26
+ "group_size": 0,
27
+ "is_integer": true,
28
+ "is_training": false,
29
+ "modules_dtype_dict": {
30
+ "int5": [
31
+ "single_transformer_blocks.23.attn.to_qkv_mlp_proj.weight"
32
+ ],
33
+ "uint4": [
34
+ "transformer_blocks.0.attn.to_q.weight",
35
+ "transformer_blocks.0.attn.to_k.weight",
36
+ "transformer_blocks.0.attn.to_v.weight",
37
+ "transformer_blocks.0.attn.to_out.0.weight",
38
+ "transformer_blocks.0.attn.add_q_proj.weight",
39
+ "transformer_blocks.0.attn.add_k_proj.weight",
40
+ "transformer_blocks.0.attn.add_v_proj.weight",
41
+ "transformer_blocks.0.attn.to_add_out.weight",
42
+ "transformer_blocks.0.ff.linear_in.weight",
43
+ "transformer_blocks.0.ff.linear_out.weight",
44
+ "transformer_blocks.0.ff_context.linear_in.weight",
45
+ "transformer_blocks.0.ff_context.linear_out.weight",
46
+ "transformer_blocks.1.attn.to_q.weight",
47
+ "transformer_blocks.1.attn.to_k.weight",
48
+ "transformer_blocks.1.attn.to_v.weight",
49
+ "transformer_blocks.1.attn.to_out.0.weight",
50
+ "transformer_blocks.1.attn.add_q_proj.weight",
51
+ "transformer_blocks.1.attn.add_k_proj.weight",
52
+ "transformer_blocks.1.attn.add_v_proj.weight",
53
+ "transformer_blocks.1.attn.to_add_out.weight",
54
+ "transformer_blocks.1.ff.linear_in.weight",
55
+ "transformer_blocks.1.ff.linear_out.weight",
56
+ "transformer_blocks.1.ff_context.linear_in.weight",
57
+ "transformer_blocks.1.ff_context.linear_out.weight",
58
+ "transformer_blocks.2.attn.to_q.weight",
59
+ "transformer_blocks.2.attn.to_k.weight",
60
+ "transformer_blocks.2.attn.to_v.weight",
61
+ "transformer_blocks.2.attn.to_out.0.weight",
62
+ "transformer_blocks.2.attn.add_q_proj.weight",
63
+ "transformer_blocks.2.attn.add_k_proj.weight",
64
+ "transformer_blocks.2.attn.add_v_proj.weight",
65
+ "transformer_blocks.2.attn.to_add_out.weight",
66
+ "transformer_blocks.2.ff.linear_in.weight",
67
+ "transformer_blocks.2.ff.linear_out.weight",
68
+ "transformer_blocks.2.ff_context.linear_in.weight",
69
+ "transformer_blocks.2.ff_context.linear_out.weight",
70
+ "transformer_blocks.3.attn.to_q.weight",
71
+ "transformer_blocks.3.attn.to_k.weight",
72
+ "transformer_blocks.3.attn.to_v.weight",
73
+ "transformer_blocks.3.attn.to_out.0.weight",
74
+ "transformer_blocks.3.attn.add_q_proj.weight",
75
+ "transformer_blocks.3.attn.add_k_proj.weight",
76
+ "transformer_blocks.3.attn.add_v_proj.weight",
77
+ "transformer_blocks.3.attn.to_add_out.weight",
78
+ "transformer_blocks.3.ff.linear_in.weight",
79
+ "transformer_blocks.3.ff.linear_out.weight",
80
+ "transformer_blocks.3.ff_context.linear_in.weight",
81
+ "transformer_blocks.3.ff_context.linear_out.weight",
82
+ "transformer_blocks.4.attn.to_q.weight",
83
+ "transformer_blocks.4.attn.to_k.weight",
84
+ "transformer_blocks.4.attn.to_v.weight",
85
+ "transformer_blocks.4.attn.to_out.0.weight",
86
+ "transformer_blocks.4.attn.add_q_proj.weight",
87
+ "transformer_blocks.4.attn.add_k_proj.weight",
88
+ "transformer_blocks.4.attn.add_v_proj.weight",
89
+ "transformer_blocks.4.attn.to_add_out.weight",
90
+ "transformer_blocks.4.ff.linear_in.weight",
91
+ "transformer_blocks.4.ff.linear_out.weight",
92
+ "transformer_blocks.4.ff_context.linear_in.weight",
93
+ "transformer_blocks.4.ff_context.linear_out.weight",
94
+ "transformer_blocks.5.attn.to_q.weight",
95
+ "transformer_blocks.5.attn.to_k.weight",
96
+ "transformer_blocks.5.attn.to_v.weight",
97
+ "transformer_blocks.5.attn.to_out.0.weight",
98
+ "transformer_blocks.5.attn.add_q_proj.weight",
99
+ "transformer_blocks.5.attn.add_k_proj.weight",
100
+ "transformer_blocks.5.attn.add_v_proj.weight",
101
+ "transformer_blocks.5.attn.to_add_out.weight",
102
+ "transformer_blocks.5.ff.linear_in.weight",
103
+ "transformer_blocks.5.ff.linear_out.weight",
104
+ "transformer_blocks.5.ff_context.linear_in.weight",
105
+ "transformer_blocks.5.ff_context.linear_out.weight",
106
+ "transformer_blocks.6.attn.to_q.weight",
107
+ "transformer_blocks.6.attn.to_k.weight",
108
+ "transformer_blocks.6.attn.to_v.weight",
109
+ "transformer_blocks.6.attn.to_out.0.weight",
110
+ "transformer_blocks.6.attn.add_q_proj.weight",
111
+ "transformer_blocks.6.attn.add_k_proj.weight",
112
+ "transformer_blocks.6.attn.add_v_proj.weight",
113
+ "transformer_blocks.6.attn.to_add_out.weight",
114
+ "transformer_blocks.6.ff.linear_in.weight",
115
+ "transformer_blocks.6.ff.linear_out.weight",
116
+ "transformer_blocks.6.ff_context.linear_in.weight",
117
+ "transformer_blocks.6.ff_context.linear_out.weight",
118
+ "transformer_blocks.7.attn.to_q.weight",
119
+ "transformer_blocks.7.attn.to_k.weight",
120
+ "transformer_blocks.7.attn.to_v.weight",
121
+ "transformer_blocks.7.attn.to_out.0.weight",
122
+ "transformer_blocks.7.attn.add_q_proj.weight",
123
+ "transformer_blocks.7.attn.add_k_proj.weight",
124
+ "transformer_blocks.7.attn.add_v_proj.weight",
125
+ "transformer_blocks.7.attn.to_add_out.weight",
126
+ "transformer_blocks.7.ff.linear_in.weight",
127
+ "transformer_blocks.7.ff.linear_out.weight",
128
+ "transformer_blocks.7.ff_context.linear_in.weight",
129
+ "transformer_blocks.7.ff_context.linear_out.weight",
130
+ "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight",
131
+ "single_transformer_blocks.0.attn.to_out.weight",
132
+ "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight",
133
+ "single_transformer_blocks.1.attn.to_out.weight",
134
+ "single_transformer_blocks.2.attn.to_qkv_mlp_proj.weight",
135
+ "single_transformer_blocks.2.attn.to_out.weight",
136
+ "single_transformer_blocks.3.attn.to_qkv_mlp_proj.weight",
137
+ "single_transformer_blocks.3.attn.to_out.weight",
138
+ "single_transformer_blocks.4.attn.to_qkv_mlp_proj.weight",
139
+ "single_transformer_blocks.4.attn.to_out.weight",
140
+ "single_transformer_blocks.5.attn.to_qkv_mlp_proj.weight",
141
+ "single_transformer_blocks.5.attn.to_out.weight",
142
+ "single_transformer_blocks.6.attn.to_qkv_mlp_proj.weight",
143
+ "single_transformer_blocks.6.attn.to_out.weight",
144
+ "single_transformer_blocks.7.attn.to_qkv_mlp_proj.weight",
145
+ "single_transformer_blocks.7.attn.to_out.weight",
146
+ "single_transformer_blocks.8.attn.to_qkv_mlp_proj.weight",
147
+ "single_transformer_blocks.8.attn.to_out.weight",
148
+ "single_transformer_blocks.9.attn.to_qkv_mlp_proj.weight",
149
+ "single_transformer_blocks.9.attn.to_out.weight",
150
+ "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight",
151
+ "single_transformer_blocks.10.attn.to_out.weight",
152
+ "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight",
153
+ "single_transformer_blocks.11.attn.to_out.weight",
154
+ "single_transformer_blocks.12.attn.to_qkv_mlp_proj.weight",
155
+ "single_transformer_blocks.12.attn.to_out.weight",
156
+ "single_transformer_blocks.13.attn.to_qkv_mlp_proj.weight",
157
+ "single_transformer_blocks.13.attn.to_out.weight",
158
+ "single_transformer_blocks.14.attn.to_qkv_mlp_proj.weight",
159
+ "single_transformer_blocks.14.attn.to_out.weight",
160
+ "single_transformer_blocks.15.attn.to_qkv_mlp_proj.weight",
161
+ "single_transformer_blocks.15.attn.to_out.weight",
162
+ "single_transformer_blocks.16.attn.to_qkv_mlp_proj.weight",
163
+ "single_transformer_blocks.16.attn.to_out.weight",
164
+ "single_transformer_blocks.17.attn.to_qkv_mlp_proj.weight",
165
+ "single_transformer_blocks.17.attn.to_out.weight",
166
+ "single_transformer_blocks.18.attn.to_qkv_mlp_proj.weight",
167
+ "single_transformer_blocks.18.attn.to_out.weight",
168
+ "single_transformer_blocks.19.attn.to_qkv_mlp_proj.weight",
169
+ "single_transformer_blocks.19.attn.to_out.weight",
170
+ "single_transformer_blocks.20.attn.to_qkv_mlp_proj.weight",
171
+ "single_transformer_blocks.20.attn.to_out.weight",
172
+ "single_transformer_blocks.21.attn.to_qkv_mlp_proj.weight",
173
+ "single_transformer_blocks.21.attn.to_out.weight",
174
+ "single_transformer_blocks.22.attn.to_qkv_mlp_proj.weight",
175
+ "single_transformer_blocks.22.attn.to_out.weight",
176
+ "single_transformer_blocks.23.attn.to_out.weight"
177
+ ]
178
+ },
179
+ "modules_quant_config": {},
180
+ "modules_to_not_convert": [
181
+ "norm_out",
182
+ "context_embedder",
183
+ "time_guidance_embed",
184
+ "double_stream_modulation_txt",
185
+ "x_embedder",
186
+ ".proj_out",
187
+ "double_stream_modulation_img",
188
+ "single_stream_modulation"
189
+ ],
190
+ "non_blocking": false,
191
+ "quant_conv": false,
192
+ "quant_embedding": false,
193
+ "quant_method": "sdnq",
194
+ "quantization_device": "cpu",
195
+ "quantized_matmul_dtype": null,
196
+ "return_device": "cpu",
197
+ "sdnq_version": "0.1.8",
198
+ "svd_rank": 32,
199
+ "svd_steps": 8,
200
+ "use_dynamic_quantization": true,
201
+ "use_grad_ckpt": true,
202
+ "use_quantized_matmul": false,
203
+ "use_quantized_matmul_conv": false,
204
+ "use_static_quantization": true,
205
+ "use_stochastic_rounding": false,
206
+ "use_svd": false,
207
+ "weights_dtype": "uint4"
208
+ },
209
+ "rope_theta": 2000,
210
+ "timestep_guidance_channels": 256
211
+ }
transformer/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f2b42aaf8acb60e2074cfe71758d45dce6949769109f88ec1206b713392de41
3
+ size 6166234776
vae/config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKLFlux2",
3
+ "_diffusers_version": "0.38.0",
4
+ "_name_or_path": "/root/.cache/huggingface/hub/models--black-forest-labs--FLUX.2-klein-9B/snapshots/92196c8e11f7b6cf2b7493e037d8c5345c559216/vae",
5
+ "act_fn": "silu",
6
+ "batch_norm_eps": 0.0001,
7
+ "batch_norm_momentum": 0.1,
8
+ "block_out_channels": [
9
+ 128,
10
+ 256,
11
+ 512,
12
+ 512
13
+ ],
14
+ "decoder_block_out_channels": null,
15
+ "down_block_types": [
16
+ "DownEncoderBlock2D",
17
+ "DownEncoderBlock2D",
18
+ "DownEncoderBlock2D",
19
+ "DownEncoderBlock2D"
20
+ ],
21
+ "force_upcast": true,
22
+ "in_channels": 3,
23
+ "latent_channels": 32,
24
+ "layers_per_block": 2,
25
+ "mid_block_add_attention": true,
26
+ "norm_num_groups": 32,
27
+ "out_channels": 3,
28
+ "patch_size": [
29
+ 2,
30
+ 2
31
+ ],
32
+ "sample_size": 1024,
33
+ "up_block_types": [
34
+ "UpDecoderBlock2D",
35
+ "UpDecoderBlock2D",
36
+ "UpDecoderBlock2D",
37
+ "UpDecoderBlock2D"
38
+ ],
39
+ "use_post_quant_conv": true,
40
+ "use_quant_conv": true
41
+ }
vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca70d2202afe6415bdbcb8793ba8cd99fd159cfe6192381504d6c4d3036e0f04
3
+ size 168120878