Hugging Face's logo

appautomaton
/
sam-3d-objects-mlx

Image-to-3D
MLX
Safetensors
SAM 3D Objects
apple-silicon
sam-3d
3d-reconstruction
gaussian-splatting
mesh
glb
Model card Files
xet
 Community
tamarher commited on
Commit
34471c8
·
verified ·
1 Parent(s): 1eb8d37

Add files using upload-large-folder tool

Browse files
Files changed (20) hide show
  1. checkpoints/conversion_metadata/slat_decoder_gs.yaml +1428 -0
  2. checkpoints/conversion_metadata/slat_decoder_gs_4.yaml +1428 -0
  3. checkpoints/conversion_metadata/slat_decoder_mesh.yaml +3 -0
  4. checkpoints/conversion_metadata/slat_generator.yaml +3 -0
  5. checkpoints/conversion_metadata/ss_decoder.yaml +1135 -0
  6. checkpoints/conversion_metadata/ss_generator.yaml +3 -0
  7. checkpoints/pipeline.yaml +106 -0
  8. checkpoints/slat_decoder_gs.safetensors +3 -0
  9. checkpoints/slat_decoder_gs.yaml +25 -0
  10. checkpoints/slat_decoder_gs_4.safetensors +3 -0
  11. checkpoints/slat_decoder_gs_4.yaml +25 -0
  12. checkpoints/slat_decoder_mesh.safetensors +3 -0
  13. checkpoints/slat_decoder_mesh.yaml +12 -0
  14. checkpoints/slat_generator.safetensors +3 -0
  15. checkpoints/slat_generator.yaml +60 -0
  16. checkpoints/ss_decoder.safetensors +3 -0
  17. checkpoints/ss_generator.safetensors +3 -0
  18. checkpoints/ss_generator.yaml +141 -0
  19. moge/conversion_metadata/model.yaml +3 -0
  20. moge/model.safetensors +3 -0
checkpoints/conversion_metadata/slat_decoder_gs.yaml ADDED
@@ -0,0 +1,1428 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ format_version: 1
2
+ source_file: weights/sam-3d-objects/checkpoints/slat_decoder_gs.ckpt
3
+ source_sha256: f8077c36a06eaf890dd93cda1937411f793dea1eb80b3dd9329f2038ba84a111
4
+ safetensors_file: slat_decoder_gs.safetensors
5
+ safetensors_files: {}
6
+ created_at_unix: 1777985860
7
+ tensor_count: 101
8
+ total_tensor_bytes: 341476480
9
+ metadata:
10
+ offset_perturbation:
11
+ $type: tensor_ref
12
+ dtype: F32
13
+ storage_key: '0'
14
+ offset_elems: 0
15
+ shape:
16
+ - 32
17
+ - 3
18
+ stride:
19
+ - 3
20
+ - 1
21
+ input_layer.weight:
22
+ $type: tensor_ref
23
+ dtype: F32
24
+ storage_key: '1'
25
+ offset_elems: 0
26
+ shape:
27
+ - 768
28
+ - 8
29
+ stride:
30
+ - 8
31
+ - 1
32
+ input_layer.bias:
33
+ $type: tensor_ref
34
+ dtype: F32
35
+ storage_key: '2'
36
+ offset_elems: 0
37
+ shape:
38
+ - 768
39
+ stride:
40
+ - 1
41
+ blocks.0.attn.to_qkv.weight:
42
+ $type: tensor_ref
43
+ dtype: F16
44
+ storage_key: '3'
45
+ offset_elems: 0
46
+ shape:
47
+ - 2304
48
+ - 768
49
+ stride:
50
+ - 768
51
+ - 1
52
+ blocks.0.attn.to_qkv.bias:
53
+ $type: tensor_ref
54
+ dtype: F16
55
+ storage_key: '4'
56
+ offset_elems: 0
57
+ shape:
58
+ - 2304
59
+ stride:
60
+ - 1
61
+ blocks.0.attn.to_out.weight:
62
+ $type: tensor_ref
63
+ dtype: F16
64
+ storage_key: '5'
65
+ offset_elems: 0
66
+ shape:
67
+ - 768
68
+ - 768
69
+ stride:
70
+ - 768
71
+ - 1
72
+ blocks.0.attn.to_out.bias:
73
+ $type: tensor_ref
74
+ dtype: F16
75
+ storage_key: '6'
76
+ offset_elems: 0
77
+ shape:
78
+ - 768
79
+ stride:
80
+ - 1
81
+ blocks.0.mlp.mlp.0.weight:
82
+ $type: tensor_ref
83
+ dtype: F16
84
+ storage_key: '7'
85
+ offset_elems: 0
86
+ shape:
87
+ - 3072
88
+ - 768
89
+ stride:
90
+ - 768
91
+ - 1
92
+ blocks.0.mlp.mlp.0.bias:
93
+ $type: tensor_ref
94
+ dtype: F16
95
+ storage_key: '8'
96
+ offset_elems: 0
97
+ shape:
98
+ - 3072
99
+ stride:
100
+ - 1
101
+ blocks.0.mlp.mlp.2.weight:
102
+ $type: tensor_ref
103
+ dtype: F16
104
+ storage_key: '9'
105
+ offset_elems: 0
106
+ shape:
107
+ - 768
108
+ - 3072
109
+ stride:
110
+ - 3072
111
+ - 1
112
+ blocks.0.mlp.mlp.2.bias:
113
+ $type: tensor_ref
114
+ dtype: F16
115
+ storage_key: '10'
116
+ offset_elems: 0
117
+ shape:
118
+ - 768
119
+ stride:
120
+ - 1
121
+ blocks.1.attn.to_qkv.weight:
122
+ $type: tensor_ref
123
+ dtype: F16
124
+ storage_key: '11'
125
+ offset_elems: 0
126
+ shape:
127
+ - 2304
128
+ - 768
129
+ stride:
130
+ - 768
131
+ - 1
132
+ blocks.1.attn.to_qkv.bias:
133
+ $type: tensor_ref
134
+ dtype: F16
135
+ storage_key: '12'
136
+ offset_elems: 0
137
+ shape:
138
+ - 2304
139
+ stride:
140
+ - 1
141
+ blocks.1.attn.to_out.weight:
142
+ $type: tensor_ref
143
+ dtype: F16
144
+ storage_key: '13'
145
+ offset_elems: 0
146
+ shape:
147
+ - 768
148
+ - 768
149
+ stride:
150
+ - 768
151
+ - 1
152
+ blocks.1.attn.to_out.bias:
153
+ $type: tensor_ref
154
+ dtype: F16
155
+ storage_key: '14'
156
+ offset_elems: 0
157
+ shape:
158
+ - 768
159
+ stride:
160
+ - 1
161
+ blocks.1.mlp.mlp.0.weight:
162
+ $type: tensor_ref
163
+ dtype: F16
164
+ storage_key: '15'
165
+ offset_elems: 0
166
+ shape:
167
+ - 3072
168
+ - 768
169
+ stride:
170
+ - 768
171
+ - 1
172
+ blocks.1.mlp.mlp.0.bias:
173
+ $type: tensor_ref
174
+ dtype: F16
175
+ storage_key: '16'
176
+ offset_elems: 0
177
+ shape:
178
+ - 3072
179
+ stride:
180
+ - 1
181
+ blocks.1.mlp.mlp.2.weight:
182
+ $type: tensor_ref
183
+ dtype: F16
184
+ storage_key: '17'
185
+ offset_elems: 0
186
+ shape:
187
+ - 768
188
+ - 3072
189
+ stride:
190
+ - 3072
191
+ - 1
192
+ blocks.1.mlp.mlp.2.bias:
193
+ $type: tensor_ref
194
+ dtype: F16
195
+ storage_key: '18'
196
+ offset_elems: 0
197
+ shape:
198
+ - 768
199
+ stride:
200
+ - 1
201
+ blocks.2.attn.to_qkv.weight:
202
+ $type: tensor_ref
203
+ dtype: F16
204
+ storage_key: '19'
205
+ offset_elems: 0
206
+ shape:
207
+ - 2304
208
+ - 768
209
+ stride:
210
+ - 768
211
+ - 1
212
+ blocks.2.attn.to_qkv.bias:
213
+ $type: tensor_ref
214
+ dtype: F16
215
+ storage_key: '20'
216
+ offset_elems: 0
217
+ shape:
218
+ - 2304
219
+ stride:
220
+ - 1
221
+ blocks.2.attn.to_out.weight:
222
+ $type: tensor_ref
223
+ dtype: F16
224
+ storage_key: '21'
225
+ offset_elems: 0
226
+ shape:
227
+ - 768
228
+ - 768
229
+ stride:
230
+ - 768
231
+ - 1
232
+ blocks.2.attn.to_out.bias:
233
+ $type: tensor_ref
234
+ dtype: F16
235
+ storage_key: '22'
236
+ offset_elems: 0
237
+ shape:
238
+ - 768
239
+ stride:
240
+ - 1
241
+ blocks.2.mlp.mlp.0.weight:
242
+ $type: tensor_ref
243
+ dtype: F16
244
+ storage_key: '23'
245
+ offset_elems: 0
246
+ shape:
247
+ - 3072
248
+ - 768
249
+ stride:
250
+ - 768
251
+ - 1
252
+ blocks.2.mlp.mlp.0.bias:
253
+ $type: tensor_ref
254
+ dtype: F16
255
+ storage_key: '24'
256
+ offset_elems: 0
257
+ shape:
258
+ - 3072
259
+ stride:
260
+ - 1
261
+ blocks.2.mlp.mlp.2.weight:
262
+ $type: tensor_ref
263
+ dtype: F16
264
+ storage_key: '25'
265
+ offset_elems: 0
266
+ shape:
267
+ - 768
268
+ - 3072
269
+ stride:
270
+ - 3072
271
+ - 1
272
+ blocks.2.mlp.mlp.2.bias:
273
+ $type: tensor_ref
274
+ dtype: F16
275
+ storage_key: '26'
276
+ offset_elems: 0
277
+ shape:
278
+ - 768
279
+ stride:
280
+ - 1
281
+ blocks.3.attn.to_qkv.weight:
282
+ $type: tensor_ref
283
+ dtype: F16
284
+ storage_key: '27'
285
+ offset_elems: 0
286
+ shape:
287
+ - 2304
288
+ - 768
289
+ stride:
290
+ - 768
291
+ - 1
292
+ blocks.3.attn.to_qkv.bias:
293
+ $type: tensor_ref
294
+ dtype: F16
295
+ storage_key: '28'
296
+ offset_elems: 0
297
+ shape:
298
+ - 2304
299
+ stride:
300
+ - 1
301
+ blocks.3.attn.to_out.weight:
302
+ $type: tensor_ref
303
+ dtype: F16
304
+ storage_key: '29'
305
+ offset_elems: 0
306
+ shape:
307
+ - 768
308
+ - 768
309
+ stride:
310
+ - 768
311
+ - 1
312
+ blocks.3.attn.to_out.bias:
313
+ $type: tensor_ref
314
+ dtype: F16
315
+ storage_key: '30'
316
+ offset_elems: 0
317
+ shape:
318
+ - 768
319
+ stride:
320
+ - 1
321
+ blocks.3.mlp.mlp.0.weight:
322
+ $type: tensor_ref
323
+ dtype: F16
324
+ storage_key: '31'
325
+ offset_elems: 0
326
+ shape:
327
+ - 3072
328
+ - 768
329
+ stride:
330
+ - 768
331
+ - 1
332
+ blocks.3.mlp.mlp.0.bias:
333
+ $type: tensor_ref
334
+ dtype: F16
335
+ storage_key: '32'
336
+ offset_elems: 0
337
+ shape:
338
+ - 3072
339
+ stride:
340
+ - 1
341
+ blocks.3.mlp.mlp.2.weight:
342
+ $type: tensor_ref
343
+ dtype: F16
344
+ storage_key: '33'
345
+ offset_elems: 0
346
+ shape:
347
+ - 768
348
+ - 3072
349
+ stride:
350
+ - 3072
351
+ - 1
352
+ blocks.3.mlp.mlp.2.bias:
353
+ $type: tensor_ref
354
+ dtype: F16
355
+ storage_key: '34'
356
+ offset_elems: 0
357
+ shape:
358
+ - 768
359
+ stride:
360
+ - 1
361
+ blocks.4.attn.to_qkv.weight:
362
+ $type: tensor_ref
363
+ dtype: F16
364
+ storage_key: '35'
365
+ offset_elems: 0
366
+ shape:
367
+ - 2304
368
+ - 768
369
+ stride:
370
+ - 768
371
+ - 1
372
+ blocks.4.attn.to_qkv.bias:
373
+ $type: tensor_ref
374
+ dtype: F16
375
+ storage_key: '36'
376
+ offset_elems: 0
377
+ shape:
378
+ - 2304
379
+ stride:
380
+ - 1
381
+ blocks.4.attn.to_out.weight:
382
+ $type: tensor_ref
383
+ dtype: F16
384
+ storage_key: '37'
385
+ offset_elems: 0
386
+ shape:
387
+ - 768
388
+ - 768
389
+ stride:
390
+ - 768
391
+ - 1
392
+ blocks.4.attn.to_out.bias:
393
+ $type: tensor_ref
394
+ dtype: F16
395
+ storage_key: '38'
396
+ offset_elems: 0
397
+ shape:
398
+ - 768
399
+ stride:
400
+ - 1
401
+ blocks.4.mlp.mlp.0.weight:
402
+ $type: tensor_ref
403
+ dtype: F16
404
+ storage_key: '39'
405
+ offset_elems: 0
406
+ shape:
407
+ - 3072
408
+ - 768
409
+ stride:
410
+ - 768
411
+ - 1
412
+ blocks.4.mlp.mlp.0.bias:
413
+ $type: tensor_ref
414
+ dtype: F16
415
+ storage_key: '40'
416
+ offset_elems: 0
417
+ shape:
418
+ - 3072
419
+ stride:
420
+ - 1
421
+ blocks.4.mlp.mlp.2.weight:
422
+ $type: tensor_ref
423
+ dtype: F16
424
+ storage_key: '41'
425
+ offset_elems: 0
426
+ shape:
427
+ - 768
428
+ - 3072
429
+ stride:
430
+ - 3072
431
+ - 1
432
+ blocks.4.mlp.mlp.2.bias:
433
+ $type: tensor_ref
434
+ dtype: F16
435
+ storage_key: '42'
436
+ offset_elems: 0
437
+ shape:
438
+ - 768
439
+ stride:
440
+ - 1
441
+ blocks.5.attn.to_qkv.weight:
442
+ $type: tensor_ref
443
+ dtype: F16
444
+ storage_key: '43'
445
+ offset_elems: 0
446
+ shape:
447
+ - 2304
448
+ - 768
449
+ stride:
450
+ - 768
451
+ - 1
452
+ blocks.5.attn.to_qkv.bias:
453
+ $type: tensor_ref
454
+ dtype: F16
455
+ storage_key: '44'
456
+ offset_elems: 0
457
+ shape:
458
+ - 2304
459
+ stride:
460
+ - 1
461
+ blocks.5.attn.to_out.weight:
462
+ $type: tensor_ref
463
+ dtype: F16
464
+ storage_key: '45'
465
+ offset_elems: 0
466
+ shape:
467
+ - 768
468
+ - 768
469
+ stride:
470
+ - 768
471
+ - 1
472
+ blocks.5.attn.to_out.bias:
473
+ $type: tensor_ref
474
+ dtype: F16
475
+ storage_key: '46'
476
+ offset_elems: 0
477
+ shape:
478
+ - 768
479
+ stride:
480
+ - 1
481
+ blocks.5.mlp.mlp.0.weight:
482
+ $type: tensor_ref
483
+ dtype: F16
484
+ storage_key: '47'
485
+ offset_elems: 0
486
+ shape:
487
+ - 3072
488
+ - 768
489
+ stride:
490
+ - 768
491
+ - 1
492
+ blocks.5.mlp.mlp.0.bias:
493
+ $type: tensor_ref
494
+ dtype: F16
495
+ storage_key: '48'
496
+ offset_elems: 0
497
+ shape:
498
+ - 3072
499
+ stride:
500
+ - 1
501
+ blocks.5.mlp.mlp.2.weight:
502
+ $type: tensor_ref
503
+ dtype: F16
504
+ storage_key: '49'
505
+ offset_elems: 0
506
+ shape:
507
+ - 768
508
+ - 3072
509
+ stride:
510
+ - 3072
511
+ - 1
512
+ blocks.5.mlp.mlp.2.bias:
513
+ $type: tensor_ref
514
+ dtype: F16
515
+ storage_key: '50'
516
+ offset_elems: 0
517
+ shape:
518
+ - 768
519
+ stride:
520
+ - 1
521
+ blocks.6.attn.to_qkv.weight:
522
+ $type: tensor_ref
523
+ dtype: F16
524
+ storage_key: '51'
525
+ offset_elems: 0
526
+ shape:
527
+ - 2304
528
+ - 768
529
+ stride:
530
+ - 768
531
+ - 1
532
+ blocks.6.attn.to_qkv.bias:
533
+ $type: tensor_ref
534
+ dtype: F16
535
+ storage_key: '52'
536
+ offset_elems: 0
537
+ shape:
538
+ - 2304
539
+ stride:
540
+ - 1
541
+ blocks.6.attn.to_out.weight:
542
+ $type: tensor_ref
543
+ dtype: F16
544
+ storage_key: '53'
545
+ offset_elems: 0
546
+ shape:
547
+ - 768
548
+ - 768
549
+ stride:
550
+ - 768
551
+ - 1
552
+ blocks.6.attn.to_out.bias:
553
+ $type: tensor_ref
554
+ dtype: F16
555
+ storage_key: '54'
556
+ offset_elems: 0
557
+ shape:
558
+ - 768
559
+ stride:
560
+ - 1
561
+ blocks.6.mlp.mlp.0.weight:
562
+ $type: tensor_ref
563
+ dtype: F16
564
+ storage_key: '55'
565
+ offset_elems: 0
566
+ shape:
567
+ - 3072
568
+ - 768
569
+ stride:
570
+ - 768
571
+ - 1
572
+ blocks.6.mlp.mlp.0.bias:
573
+ $type: tensor_ref
574
+ dtype: F16
575
+ storage_key: '56'
576
+ offset_elems: 0
577
+ shape:
578
+ - 3072
579
+ stride:
580
+ - 1
581
+ blocks.6.mlp.mlp.2.weight:
582
+ $type: tensor_ref
583
+ dtype: F16
584
+ storage_key: '57'
585
+ offset_elems: 0
586
+ shape:
587
+ - 768
588
+ - 3072
589
+ stride:
590
+ - 3072
591
+ - 1
592
+ blocks.6.mlp.mlp.2.bias:
593
+ $type: tensor_ref
594
+ dtype: F16
595
+ storage_key: '58'
596
+ offset_elems: 0
597
+ shape:
598
+ - 768
599
+ stride:
600
+ - 1
601
+ blocks.7.attn.to_qkv.weight:
602
+ $type: tensor_ref
603
+ dtype: F16
604
+ storage_key: '59'
605
+ offset_elems: 0
606
+ shape:
607
+ - 2304
608
+ - 768
609
+ stride:
610
+ - 768
611
+ - 1
612
+ blocks.7.attn.to_qkv.bias:
613
+ $type: tensor_ref
614
+ dtype: F16
615
+ storage_key: '60'
616
+ offset_elems: 0
617
+ shape:
618
+ - 2304
619
+ stride:
620
+ - 1
621
+ blocks.7.attn.to_out.weight:
622
+ $type: tensor_ref
623
+ dtype: F16
624
+ storage_key: '61'
625
+ offset_elems: 0
626
+ shape:
627
+ - 768
628
+ - 768
629
+ stride:
630
+ - 768
631
+ - 1
632
+ blocks.7.attn.to_out.bias:
633
+ $type: tensor_ref
634
+ dtype: F16
635
+ storage_key: '62'
636
+ offset_elems: 0
637
+ shape:
638
+ - 768
639
+ stride:
640
+ - 1
641
+ blocks.7.mlp.mlp.0.weight:
642
+ $type: tensor_ref
643
+ dtype: F16
644
+ storage_key: '63'
645
+ offset_elems: 0
646
+ shape:
647
+ - 3072
648
+ - 768
649
+ stride:
650
+ - 768
651
+ - 1
652
+ blocks.7.mlp.mlp.0.bias:
653
+ $type: tensor_ref
654
+ dtype: F16
655
+ storage_key: '64'
656
+ offset_elems: 0
657
+ shape:
658
+ - 3072
659
+ stride:
660
+ - 1
661
+ blocks.7.mlp.mlp.2.weight:
662
+ $type: tensor_ref
663
+ dtype: F16
664
+ storage_key: '65'
665
+ offset_elems: 0
666
+ shape:
667
+ - 768
668
+ - 3072
669
+ stride:
670
+ - 3072
671
+ - 1
672
+ blocks.7.mlp.mlp.2.bias:
673
+ $type: tensor_ref
674
+ dtype: F16
675
+ storage_key: '66'
676
+ offset_elems: 0
677
+ shape:
678
+ - 768
679
+ stride:
680
+ - 1
681
+ blocks.8.attn.to_qkv.weight:
682
+ $type: tensor_ref
683
+ dtype: F16
684
+ storage_key: '67'
685
+ offset_elems: 0
686
+ shape:
687
+ - 2304
688
+ - 768
689
+ stride:
690
+ - 768
691
+ - 1
692
+ blocks.8.attn.to_qkv.bias:
693
+ $type: tensor_ref
694
+ dtype: F16
695
+ storage_key: '68'
696
+ offset_elems: 0
697
+ shape:
698
+ - 2304
699
+ stride:
700
+ - 1
701
+ blocks.8.attn.to_out.weight:
702
+ $type: tensor_ref
703
+ dtype: F16
704
+ storage_key: '69'
705
+ offset_elems: 0
706
+ shape:
707
+ - 768
708
+ - 768
709
+ stride:
710
+ - 768
711
+ - 1
712
+ blocks.8.attn.to_out.bias:
713
+ $type: tensor_ref
714
+ dtype: F16
715
+ storage_key: '70'
716
+ offset_elems: 0
717
+ shape:
718
+ - 768
719
+ stride:
720
+ - 1
721
+ blocks.8.mlp.mlp.0.weight:
722
+ $type: tensor_ref
723
+ dtype: F16
724
+ storage_key: '71'
725
+ offset_elems: 0
726
+ shape:
727
+ - 3072
728
+ - 768
729
+ stride:
730
+ - 768
731
+ - 1
732
+ blocks.8.mlp.mlp.0.bias:
733
+ $type: tensor_ref
734
+ dtype: F16
735
+ storage_key: '72'
736
+ offset_elems: 0
737
+ shape:
738
+ - 3072
739
+ stride:
740
+ - 1
741
+ blocks.8.mlp.mlp.2.weight:
742
+ $type: tensor_ref
743
+ dtype: F16
744
+ storage_key: '73'
745
+ offset_elems: 0
746
+ shape:
747
+ - 768
748
+ - 3072
749
+ stride:
750
+ - 3072
751
+ - 1
752
+ blocks.8.mlp.mlp.2.bias:
753
+ $type: tensor_ref
754
+ dtype: F16
755
+ storage_key: '74'
756
+ offset_elems: 0
757
+ shape:
758
+ - 768
759
+ stride:
760
+ - 1
761
+ blocks.9.attn.to_qkv.weight:
762
+ $type: tensor_ref
763
+ dtype: F16
764
+ storage_key: '75'
765
+ offset_elems: 0
766
+ shape:
767
+ - 2304
768
+ - 768
769
+ stride:
770
+ - 768
771
+ - 1
772
+ blocks.9.attn.to_qkv.bias:
773
+ $type: tensor_ref
774
+ dtype: F16
775
+ storage_key: '76'
776
+ offset_elems: 0
777
+ shape:
778
+ - 2304
779
+ stride:
780
+ - 1
781
+ blocks.9.attn.to_out.weight:
782
+ $type: tensor_ref
783
+ dtype: F16
784
+ storage_key: '77'
785
+ offset_elems: 0
786
+ shape:
787
+ - 768
788
+ - 768
789
+ stride:
790
+ - 768
791
+ - 1
792
+ blocks.9.attn.to_out.bias:
793
+ $type: tensor_ref
794
+ dtype: F16
795
+ storage_key: '78'
796
+ offset_elems: 0
797
+ shape:
798
+ - 768
799
+ stride:
800
+ - 1
801
+ blocks.9.mlp.mlp.0.weight:
802
+ $type: tensor_ref
803
+ dtype: F16
804
+ storage_key: '79'
805
+ offset_elems: 0
806
+ shape:
807
+ - 3072
808
+ - 768
809
+ stride:
810
+ - 768
811
+ - 1
812
+ blocks.9.mlp.mlp.0.bias:
813
+ $type: tensor_ref
814
+ dtype: F16
815
+ storage_key: '80'
816
+ offset_elems: 0
817
+ shape:
818
+ - 3072
819
+ stride:
820
+ - 1
821
+ blocks.9.mlp.mlp.2.weight:
822
+ $type: tensor_ref
823
+ dtype: F16
824
+ storage_key: '81'
825
+ offset_elems: 0
826
+ shape:
827
+ - 768
828
+ - 3072
829
+ stride:
830
+ - 3072
831
+ - 1
832
+ blocks.9.mlp.mlp.2.bias:
833
+ $type: tensor_ref
834
+ dtype: F16
835
+ storage_key: '82'
836
+ offset_elems: 0
837
+ shape:
838
+ - 768
839
+ stride:
840
+ - 1
841
+ blocks.10.attn.to_qkv.weight:
842
+ $type: tensor_ref
843
+ dtype: F16
844
+ storage_key: '83'
845
+ offset_elems: 0
846
+ shape:
847
+ - 2304
848
+ - 768
849
+ stride:
850
+ - 768
851
+ - 1
852
+ blocks.10.attn.to_qkv.bias:
853
+ $type: tensor_ref
854
+ dtype: F16
855
+ storage_key: '84'
856
+ offset_elems: 0
857
+ shape:
858
+ - 2304
859
+ stride:
860
+ - 1
861
+ blocks.10.attn.to_out.weight:
862
+ $type: tensor_ref
863
+ dtype: F16
864
+ storage_key: '85'
865
+ offset_elems: 0
866
+ shape:
867
+ - 768
868
+ - 768
869
+ stride:
870
+ - 768
871
+ - 1
872
+ blocks.10.attn.to_out.bias:
873
+ $type: tensor_ref
874
+ dtype: F16
875
+ storage_key: '86'
876
+ offset_elems: 0
877
+ shape:
878
+ - 768
879
+ stride:
880
+ - 1
881
+ blocks.10.mlp.mlp.0.weight:
882
+ $type: tensor_ref
883
+ dtype: F16
884
+ storage_key: '87'
885
+ offset_elems: 0
886
+ shape:
887
+ - 3072
888
+ - 768
889
+ stride:
890
+ - 768
891
+ - 1
892
+ blocks.10.mlp.mlp.0.bias:
893
+ $type: tensor_ref
894
+ dtype: F16
895
+ storage_key: '88'
896
+ offset_elems: 0
897
+ shape:
898
+ - 3072
899
+ stride:
900
+ - 1
901
+ blocks.10.mlp.mlp.2.weight:
902
+ $type: tensor_ref
903
+ dtype: F16
904
+ storage_key: '89'
905
+ offset_elems: 0
906
+ shape:
907
+ - 768
908
+ - 3072
909
+ stride:
910
+ - 3072
911
+ - 1
912
+ blocks.10.mlp.mlp.2.bias:
913
+ $type: tensor_ref
914
+ dtype: F16
915
+ storage_key: '90'
916
+ offset_elems: 0
917
+ shape:
918
+ - 768
919
+ stride:
920
+ - 1
921
+ blocks.11.attn.to_qkv.weight:
922
+ $type: tensor_ref
923
+ dtype: F16
924
+ storage_key: '91'
925
+ offset_elems: 0
926
+ shape:
927
+ - 2304
928
+ - 768
929
+ stride:
930
+ - 768
931
+ - 1
932
+ blocks.11.attn.to_qkv.bias:
933
+ $type: tensor_ref
934
+ dtype: F16
935
+ storage_key: '92'
936
+ offset_elems: 0
937
+ shape:
938
+ - 2304
939
+ stride:
940
+ - 1
941
+ blocks.11.attn.to_out.weight:
942
+ $type: tensor_ref
943
+ dtype: F16
944
+ storage_key: '93'
945
+ offset_elems: 0
946
+ shape:
947
+ - 768
948
+ - 768
949
+ stride:
950
+ - 768
951
+ - 1
952
+ blocks.11.attn.to_out.bias:
953
+ $type: tensor_ref
954
+ dtype: F16
955
+ storage_key: '94'
956
+ offset_elems: 0
957
+ shape:
958
+ - 768
959
+ stride:
960
+ - 1
961
+ blocks.11.mlp.mlp.0.weight:
962
+ $type: tensor_ref
963
+ dtype: F16
964
+ storage_key: '95'
965
+ offset_elems: 0
966
+ shape:
967
+ - 3072
968
+ - 768
969
+ stride:
970
+ - 768
971
+ - 1
972
+ blocks.11.mlp.mlp.0.bias:
973
+ $type: tensor_ref
974
+ dtype: F16
975
+ storage_key: '96'
976
+ offset_elems: 0
977
+ shape:
978
+ - 3072
979
+ stride:
980
+ - 1
981
+ blocks.11.mlp.mlp.2.weight:
982
+ $type: tensor_ref
983
+ dtype: F16
984
+ storage_key: '97'
985
+ offset_elems: 0
986
+ shape:
987
+ - 768
988
+ - 3072
989
+ stride:
990
+ - 3072
991
+ - 1
992
+ blocks.11.mlp.mlp.2.bias:
993
+ $type: tensor_ref
994
+ dtype: F16
995
+ storage_key: '98'
996
+ offset_elems: 0
997
+ shape:
998
+ - 768
999
+ stride:
1000
+ - 1
1001
+ out_layer.weight:
1002
+ $type: tensor_ref
1003
+ dtype: F32
1004
+ storage_key: '99'
1005
+ offset_elems: 0
1006
+ shape:
1007
+ - 448
1008
+ - 768
1009
+ stride:
1010
+ - 768
1011
+ - 1
1012
+ out_layer.bias:
1013
+ $type: tensor_ref
1014
+ dtype: F32
1015
+ storage_key: '100'
1016
+ offset_elems: 0
1017
+ shape:
1018
+ - 448
1019
+ stride:
1020
+ - 1
1021
+ security:
1022
+ objects: []
1023
+ calls: []
1024
+ tensors:
1025
+ - name: blocks.0.attn.to_out.bias
1026
+ dtype: F32
1027
+ shape: [768]
1028
+ sha256: e6d4bda093264dc25bdae341914ba7b30b67a8cc27cbcae4f207a80180a90264
1029
+ - name: blocks.0.attn.to_out.weight
1030
+ dtype: F32
1031
+ shape: [768, 768]
1032
+ sha256: 1f3def8f8ee4ae063df6544adb441345537525e0ccad30adb3c646e711216458
1033
+ - name: blocks.0.attn.to_qkv.bias
1034
+ dtype: F32
1035
+ shape: [2304]
1036
+ sha256: 014934a6951b8b51df3202720679d76cae806abad3e69ee45b514a7afa8f820b
1037
+ - name: blocks.0.attn.to_qkv.weight
1038
+ dtype: F32
1039
+ shape: [2304, 768]
1040
+ sha256: b15c4e677e32120d0d1da9095737b1a5c46f1efeb0fd96259b5e60897f125581
1041
+ - name: blocks.0.mlp.mlp.0.bias
1042
+ dtype: F32
1043
+ shape: [3072]
1044
+ sha256: d00cb59c9b5701bdf256718e289d51f0f4f222e7a4166084b6e15ba7aca08ec4
1045
+ - name: blocks.0.mlp.mlp.0.weight
1046
+ dtype: F32
1047
+ shape: [3072, 768]
1048
+ sha256: 970db3c6bca917fb42ce32d925224f31d6d9d3cdf80ccac6ab2fc5539fc2788d
1049
+ - name: blocks.0.mlp.mlp.2.bias
1050
+ dtype: F32
1051
+ shape: [768]
1052
+ sha256: aade0d7bfeea09059fc602e324874c201698a4ccf3f8243b94d364e2c2555ee5
1053
+ - name: blocks.0.mlp.mlp.2.weight
1054
+ dtype: F32
1055
+ shape: [768, 3072]
1056
+ sha256: dff1a4f13ae7f282dec50d75b7fee624960a86551556d0e7bba623014ef0c738
1057
+ - name: blocks.1.attn.to_out.bias
1058
+ dtype: F32
1059
+ shape: [768]
1060
+ sha256: b357757de05e75f8c46aef77c675070a07f3203fc29af971608fb8548c2c6a93
1061
+ - name: blocks.1.attn.to_out.weight
1062
+ dtype: F32
1063
+ shape: [768, 768]
1064
+ sha256: de70952a977faeadc8d8abe69b01609c409bea47f8d3b86654c98027345342fa
1065
+ - name: blocks.1.attn.to_qkv.bias
1066
+ dtype: F32
1067
+ shape: [2304]
1068
+ sha256: 2b17b672d6bb5150922b1328e6843554b0153da05f9915cb190e562fe3933f13
1069
+ - name: blocks.1.attn.to_qkv.weight
1070
+ dtype: F32
1071
+ shape: [2304, 768]
1072
+ sha256: 6226a65699fdbb370e3ebd8c835d84af06dfef63f88cbd14cea3f1b2cf99c6b4
1073
+ - name: blocks.1.mlp.mlp.0.bias
1074
+ dtype: F32
1075
+ shape: [3072]
1076
+ sha256: cbd81527d5b0ba5cff89f0810960a68fd323793a41ce4866d6793ffb87951623
1077
+ - name: blocks.1.mlp.mlp.0.weight
1078
+ dtype: F32
1079
+ shape: [3072, 768]
1080
+ sha256: 572b948fb1056dccf125ff10181bcc2276076243273c534d268638c973666493
1081
+ - name: blocks.1.mlp.mlp.2.bias
1082
+ dtype: F32
1083
+ shape: [768]
1084
+ sha256: ce97cb9e8be967de44a4b864c2acac107a063fced08582090c2095139137c32d
1085
+ - name: blocks.1.mlp.mlp.2.weight
1086
+ dtype: F32
1087
+ shape: [768, 3072]
1088
+ sha256: 1d27d82c2208a7adcc153d31bab0dfe714ff71eada217fa3a2721b5f93ef7240
1089
+ - name: blocks.10.attn.to_out.bias
1090
+ dtype: F32
1091
+ shape: [768]
1092
+ sha256: 0cdafc6a6d79a00c550c67526a468729b198ed08251e37534e7d05e164e09600
1093
+ - name: blocks.10.attn.to_out.weight
1094
+ dtype: F32
1095
+ shape: [768, 768]
1096
+ sha256: 51afb71969d9ec8af47f8c7835012b1f11a0bb59098d94d395999d2c20ce5adb
1097
+ - name: blocks.10.attn.to_qkv.bias
1098
+ dtype: F32
1099
+ shape: [2304]
1100
+ sha256: b6b9322399c5a4bbef2bfce983b8b2338931e5077a8ac07cdf89337a05788609
1101
+ - name: blocks.10.attn.to_qkv.weight
1102
+ dtype: F32
1103
+ shape: [2304, 768]
1104
+ sha256: f4bb5ca71024b2d7eb7c72fff83c6af798c2bc267552374088b5f82012655629
1105
+ - name: blocks.10.mlp.mlp.0.bias
1106
+ dtype: F32
1107
+ shape: [3072]
1108
+ sha256: 013c7ebc43eaabe42be9887b0e11e0b0203c595e4cc08439caaea1b466b673f8
1109
+ - name: blocks.10.mlp.mlp.0.weight
1110
+ dtype: F32
1111
+ shape: [3072, 768]
1112
+ sha256: 3ba743174521436b21db03d94ac474d3a4a25945903966940d3b3e731683712d
1113
+ - name: blocks.10.mlp.mlp.2.bias
1114
+ dtype: F32
1115
+ shape: [768]
1116
+ sha256: 0398c01842566bf2da9c16fbfa61b0512bb11409530177aa8559520dc7f49ec1
1117
+ - name: blocks.10.mlp.mlp.2.weight
1118
+ dtype: F32
1119
+ shape: [768, 3072]
1120
+ sha256: d49ed14a21d08249e8fa917a9f5d1322a27fb458427bed9db6d05989d06996fa
1121
+ - name: blocks.11.attn.to_out.bias
1122
+ dtype: F32
1123
+ shape: [768]
1124
+ sha256: 78556415562c03ce6f98235ccce7773e24441dc56848295f6467e451cfa5d83f
1125
+ - name: blocks.11.attn.to_out.weight
1126
+ dtype: F32
1127
+ shape: [768, 768]
1128
+ sha256: 25279c8174cb74c3678a724c6755670b567a84e7d36b2e0f51ef04d6dd171098
1129
+ - name: blocks.11.attn.to_qkv.bias
1130
+ dtype: F32
1131
+ shape: [2304]
1132
+ sha256: c24810f1c01c8f9e6ff7a3ca349734f804c014734a4f285e31b722a71f165ac8
1133
+ - name: blocks.11.attn.to_qkv.weight
1134
+ dtype: F32
1135
+ shape: [2304, 768]
1136
+ sha256: e267a3d7174f88e6fc0498fa1ae67e790caa690157df77be28fa342bf71f0eef
1137
+ - name: blocks.11.mlp.mlp.0.bias
1138
+ dtype: F32
1139
+ shape: [3072]
1140
+ sha256: 5c31b7af1b0d98aaa32b1443ec743d0c8d237bb334fca23f409ac7a5f30a74f2
1141
+ - name: blocks.11.mlp.mlp.0.weight
1142
+ dtype: F32
1143
+ shape: [3072, 768]
1144
+ sha256: 338d1b98a4da1e232ef9bde8232fa92b50b6d98dc86949079248db07606edcd8
1145
+ - name: blocks.11.mlp.mlp.2.bias
1146
+ dtype: F32
1147
+ shape: [768]
1148
+ sha256: 29a9badc365ccc1804276c3bb6d3ff9fa4b72cab3ea21617e98d39f52cb5df2a
1149
+ - name: blocks.11.mlp.mlp.2.weight
1150
+ dtype: F32
1151
+ shape: [768, 3072]
1152
+ sha256: e3fdafd8f36d5d5f93c89c276e93a03c5ef945ceb7a41d008e4494ed0acdd5e7
1153
+ - name: blocks.2.attn.to_out.bias
1154
+ dtype: F32
1155
+ shape: [768]
1156
+ sha256: fcf44463fc694e788b5563c5a239facdd699931d9a842be593a1acf58c497172
1157
+ - name: blocks.2.attn.to_out.weight
1158
+ dtype: F32
1159
+ shape: [768, 768]
1160
+ sha256: 8c4a4ad9bd60077b31c40ae1ad6316e9df9d63c62949e4631a21cbe2a105fe92
1161
+ - name: blocks.2.attn.to_qkv.bias
1162
+ dtype: F32
1163
+ shape: [2304]
1164
+ sha256: 4f77ab71e841dbe5cd8b15d258c416dc3c10f77314a05dedef27cebe5e6af380
1165
+ - name: blocks.2.attn.to_qkv.weight
1166
+ dtype: F32
1167
+ shape: [2304, 768]
1168
+ sha256: 1684f89b9f39558596a1453e847b1feb9495e9d040251699f15dd3a6b4c818a6
1169
+ - name: blocks.2.mlp.mlp.0.bias
1170
+ dtype: F32
1171
+ shape: [3072]
1172
+ sha256: 4d3412a469a9cc576c74ba3ddc39d8e98d9d86657415bf8efdebf75fcd94c0be
1173
+ - name: blocks.2.mlp.mlp.0.weight
1174
+ dtype: F32
1175
+ shape: [3072, 768]
1176
+ sha256: d6f4d0145ed91f378cd1f92d012c6377412346eb359db397372beea01dcf56fb
1177
+ - name: blocks.2.mlp.mlp.2.bias
1178
+ dtype: F32
1179
+ shape: [768]
1180
+ sha256: 383f601f84faa96de7eee7982c82e0e15cadb8b6af6a41de768feccc3f150823
1181
+ - name: blocks.2.mlp.mlp.2.weight
1182
+ dtype: F32
1183
+ shape: [768, 3072]
1184
+ sha256: 535823ea39db0eeb440ca21e128e9bbf1e0e44e0f629b81937d5d9220df59c5b
1185
+ - name: blocks.3.attn.to_out.bias
1186
+ dtype: F32
1187
+ shape: [768]
1188
+ sha256: f0890b63a5744ca678a2e8e9dee4dfcad0f28a77a565828a42e19ee13661d783
1189
+ - name: blocks.3.attn.to_out.weight
1190
+ dtype: F32
1191
+ shape: [768, 768]
1192
+ sha256: 480919acf28fc7a4f8b302774927dc921ab5adcd8ff16ee1fdba7f6617f51636
1193
+ - name: blocks.3.attn.to_qkv.bias
1194
+ dtype: F32
1195
+ shape: [2304]
1196
+ sha256: 22d68972286fb99409ed82a79d6f6f2ed1a78e23c395f9c370ca6a3b5d9026f7
1197
+ - name: blocks.3.attn.to_qkv.weight
1198
+ dtype: F32
1199
+ shape: [2304, 768]
1200
+ sha256: ddb6f852b36dc2582cb832cdd7ef2d1b8a49393196e2d9589a53820ef93e4c06
1201
+ - name: blocks.3.mlp.mlp.0.bias
1202
+ dtype: F32
1203
+ shape: [3072]
1204
+ sha256: f6246728c183c40723f071e18294c105bdfde003470f5ccfb9ba4d1b65e90a2c
1205
+ - name: blocks.3.mlp.mlp.0.weight
1206
+ dtype: F32
1207
+ shape: [3072, 768]
1208
+ sha256: 9ff6f926f38506ced8645dd140d0b51c71572fb8fb0ab27e8a2ed40c66409e52
1209
+ - name: blocks.3.mlp.mlp.2.bias
1210
+ dtype: F32
1211
+ shape: [768]
1212
+ sha256: 9052f10c63a1d1e49a05c8b9a40f1da9e9d153626610b11169f38d9e6b402f33
1213
+ - name: blocks.3.mlp.mlp.2.weight
1214
+ dtype: F32
1215
+ shape: [768, 3072]
1216
+ sha256: cf38ad2af18dfdf4babb19c6f271efbb8a59e87d03b7598aa2d713c795ca65e7
1217
+ - name: blocks.4.attn.to_out.bias
1218
+ dtype: F32
1219
+ shape: [768]
1220
+ sha256: 3273be9b2b4774c361fe18297dd77d4fb783d8050629bbeb9797cd5e35cf47ab
1221
+ - name: blocks.4.attn.to_out.weight
1222
+ dtype: F32
1223
+ shape: [768, 768]
1224
+ sha256: cbc06d4d122d58ad5922a6eafcd2038be8d9a0b877730e0c8c5b143198c2d24d
1225
+ - name: blocks.4.attn.to_qkv.bias
1226
+ dtype: F32
1227
+ shape: [2304]
1228
+ sha256: 0b789fe1bd72f73279e859f61c5eb87c9d691a628f691dc5c45893dcc6d39840
1229
+ - name: blocks.4.attn.to_qkv.weight
1230
+ dtype: F32
1231
+ shape: [2304, 768]
1232
+ sha256: 8601044d7e69b6e008ad04a4baf5229830317e240a45431b5e7b2d8b594d55b6
1233
+ - name: blocks.4.mlp.mlp.0.bias
1234
+ dtype: F32
1235
+ shape: [3072]
1236
+ sha256: 6203a6146106a29f88f7e3d9adf740c27a073f5ffc34b75627fe7ea060fdc9d8
1237
+ - name: blocks.4.mlp.mlp.0.weight
1238
+ dtype: F32
1239
+ shape: [3072, 768]
1240
+ sha256: 07abbf41a1ea7044dbfc78f693c1ebd05c3c031de57b12a133ca81d1cc826a3f
1241
+ - name: blocks.4.mlp.mlp.2.bias
1242
+ dtype: F32
1243
+ shape: [768]
1244
+ sha256: 99fd6f836976f491a4f76e6afc283e9271b1484473528a1888a3735d693054ec
1245
+ - name: blocks.4.mlp.mlp.2.weight
1246
+ dtype: F32
1247
+ shape: [768, 3072]
1248
+ sha256: b31e8206bfea3c560752acfded37ad2251653f879c24a7bbc6f0af809ae7775e
1249
+ - name: blocks.5.attn.to_out.bias
1250
+ dtype: F32
1251
+ shape: [768]
1252
+ sha256: 2cacc8f9f508f12fb3a9da278aae459a70cd195c30f4937d9c4c6e13c73f33bb
1253
+ - name: blocks.5.attn.to_out.weight
1254
+ dtype: F32
1255
+ shape: [768, 768]
1256
+ sha256: 57c1db0e72fc2f3a7edf09496cb66f5a38fd35a15e96cc3d77fefdfb9a9afacb
1257
+ - name: blocks.5.attn.to_qkv.bias
1258
+ dtype: F32
1259
+ shape: [2304]
1260
+ sha256: 384e1bcc2fdb5ea9b25652b95cc0c66af0d778cf5a7f3f6cb49948893c16873d
1261
+ - name: blocks.5.attn.to_qkv.weight
1262
+ dtype: F32
1263
+ shape: [2304, 768]
1264
+ sha256: 3ab2c2cf13dbdf5b4bbdc12d97839d864296cc38db44dd84987c4e77a9ca1455
1265
+ - name: blocks.5.mlp.mlp.0.bias
1266
+ dtype: F32
1267
+ shape: [3072]
1268
+ sha256: 8ad4fa2cf6c5d69eda566a6671f70b54347f8838a5a1fb3fd6712ac6312166bf
1269
+ - name: blocks.5.mlp.mlp.0.weight
1270
+ dtype: F32
1271
+ shape: [3072, 768]
1272
+ sha256: 7ecb83da54253a17f18e2ce7d3b29c3f5b305140171a94da890fea2d6b105baa
1273
+ - name: blocks.5.mlp.mlp.2.bias
1274
+ dtype: F32
1275
+ shape: [768]
1276
+ sha256: a15cf27643006d82723ca8db2a541304d55f4a495e44ab8ec674e2283dd5340e
1277
+ - name: blocks.5.mlp.mlp.2.weight
1278
+ dtype: F32
1279
+ shape: [768, 3072]
1280
+ sha256: b24f2b1dda25d6388b2ae413c14731405f473c62edaa7c0f7464702a07f0435c
1281
+ - name: blocks.6.attn.to_out.bias
1282
+ dtype: F32
1283
+ shape: [768]
1284
+ sha256: c7d7029547390271965e5e6a99dc6a8596ff13b121175315b1d6d4cf4fe01a1c
1285
+ - name: blocks.6.attn.to_out.weight
1286
+ dtype: F32
1287
+ shape: [768, 768]
1288
+ sha256: acfd054c5ee200d2a3befb95fcbd40c82f50a0818f5adf6a943176666fc14228
1289
+ - name: blocks.6.attn.to_qkv.bias
1290
+ dtype: F32
1291
+ shape: [2304]
1292
+ sha256: 764c90c34c009d81e7115e0282317d8e1a2a7fe154b30760aa76c44cb3d36994
1293
+ - name: blocks.6.attn.to_qkv.weight
1294
+ dtype: F32
1295
+ shape: [2304, 768]
1296
+ sha256: 915c7937b080e824d4aabc635ec52a55e673e02fff2dc0d188ead0afb86ff15a
1297
+ - name: blocks.6.mlp.mlp.0.bias
1298
+ dtype: F32
1299
+ shape: [3072]
1300
+ sha256: 1e4a1963af505cc023dc498ce30252ad9d3b8d3a92e747dda2ee1a58e0ffc867
1301
+ - name: blocks.6.mlp.mlp.0.weight
1302
+ dtype: F32
1303
+ shape: [3072, 768]
1304
+ sha256: e38e108f044153d2b0b2cfd70416fd78d132862d425968243c672bd1a26c4d2c
1305
+ - name: blocks.6.mlp.mlp.2.bias
1306
+ dtype: F32
1307
+ shape: [768]
1308
+ sha256: 6c4395df258d6226e28f406ec4ae9db18434b31a83eda307a74e03c605ee279a
1309
+ - name: blocks.6.mlp.mlp.2.weight
1310
+ dtype: F32
1311
+ shape: [768, 3072]
1312
+ sha256: 0ed51bf81d47538959110ac9151302ab3dc1aede2a4eeec54b219b22ebd77e8c
1313
+ - name: blocks.7.attn.to_out.bias
1314
+ dtype: F32
1315
+ shape: [768]
1316
+ sha256: 1d6870adcc72e38212abf380923af1fa867fac3f5c1cbe399af889c1ed7731b2
1317
+ - name: blocks.7.attn.to_out.weight
1318
+ dtype: F32
1319
+ shape: [768, 768]
1320
+ sha256: a2c168863f66026fdf0b4d8ef143cca78995b5b6b2e5360c360418870c75a211
1321
+ - name: blocks.7.attn.to_qkv.bias
1322
+ dtype: F32
1323
+ shape: [2304]
1324
+ sha256: e3f0a9033c9a67be70cbff6ba647559d763fdfb81d29d603b20d9b9452a774c4
1325
+ - name: blocks.7.attn.to_qkv.weight
1326
+ dtype: F32
1327
+ shape: [2304, 768]
1328
+ sha256: 076a32f33cdb1654bf95e36b78f1975d2fcfc8791d1dd4fb5ca34eb67c49cd15
1329
+ - name: blocks.7.mlp.mlp.0.bias
1330
+ dtype: F32
1331
+ shape: [3072]
1332
+ sha256: da31a71e67448923dbb5b996cec3ae9125dd89fc71064653dddff6738b2b37fc
1333
+ - name: blocks.7.mlp.mlp.0.weight
1334
+ dtype: F32
1335
+ shape: [3072, 768]
1336
+ sha256: 06b6de66d545a1b1f250e178db81afc2d220619882ef83bd5478792250001868
1337
+ - name: blocks.7.mlp.mlp.2.bias
1338
+ dtype: F32
1339
+ shape: [768]
1340
+ sha256: f767c02e187422fd3fe1c446a72701c7ecba50fbf29a474484c21517cc52cc6d
1341
+ - name: blocks.7.mlp.mlp.2.weight
1342
+ dtype: F32
1343
+ shape: [768, 3072]
1344
+ sha256: 76c77256a3c4762a8fba6ac225ae87c27d0b6a65b224a18d8d1eaa88f8fa5e0c
1345
+ - name: blocks.8.attn.to_out.bias
1346
+ dtype: F32
1347
+ shape: [768]
1348
+ sha256: 507a313c063e20f12e3b2bb9045df454cf193c0f7fb83b8aff36021b221f2212
1349
+ - name: blocks.8.attn.to_out.weight
1350
+ dtype: F32
1351
+ shape: [768, 768]
1352
+ sha256: 6c7df1bb2aa36c6940b8c12c27b267c5a8db2215e132d7673934d97c84982f64
1353
+ - name: blocks.8.attn.to_qkv.bias
1354
+ dtype: F32
1355
+ shape: [2304]
1356
+ sha256: d1e0e937b708949d7e968e4a69bcd471b1f24df394a02c68aa732f375ad3b924
1357
+ - name: blocks.8.attn.to_qkv.weight
1358
+ dtype: F32
1359
+ shape: [2304, 768]
1360
+ sha256: cfa59dcb910b356209c705ce0355c23083c24be2d255ad9deaa7a32f26821153
1361
+ - name: blocks.8.mlp.mlp.0.bias
1362
+ dtype: F32
1363
+ shape: [3072]
1364
+ sha256: 1b6be5b7d06a8954f57a86c593196d7763942b28bd2f648e15da26336d2b7926
1365
+ - name: blocks.8.mlp.mlp.0.weight
1366
+ dtype: F32
1367
+ shape: [3072, 768]
1368
+ sha256: 9aeace2c9674e20f049dc457d8ff4bea6bdf79915798332178b33fad06c58704
1369
+ - name: blocks.8.mlp.mlp.2.bias
1370
+ dtype: F32
1371
+ shape: [768]
1372
+ sha256: 8e41e8e0c2cb073ed5a1a07beb801f23f198008b926637a664268278fa0780d5
1373
+ - name: blocks.8.mlp.mlp.2.weight
1374
+ dtype: F32
1375
+ shape: [768, 3072]
1376
+ sha256: 26d6493ea4ff8fb64b41385fff67aed5789c27ceecb47debe3513dcb524593a0
1377
+ - name: blocks.9.attn.to_out.bias
1378
+ dtype: F32
1379
+ shape: [768]
1380
+ sha256: 849f11299af8c72b4db3fce2ecd144908a6b00c31bafa498f9a0ddc1a2bb030c
1381
+ - name: blocks.9.attn.to_out.weight
1382
+ dtype: F32
1383
+ shape: [768, 768]
1384
+ sha256: fb09107ee00d76034ccdfb16a21545245815d8924aad4642805d884ef5adb5c3
1385
+ - name: blocks.9.attn.to_qkv.bias
1386
+ dtype: F32
1387
+ shape: [2304]
1388
+ sha256: 583e40091afeb9d5132858ebef30325ca16e2a06b379b6b1a6c7cbbe31625539
1389
+ - name: blocks.9.attn.to_qkv.weight
1390
+ dtype: F32
1391
+ shape: [2304, 768]
1392
+ sha256: 524aa608239d56453b0b2d915eff23ca1ea79d85760e8e49879bfb56cdc41d92
1393
+ - name: blocks.9.mlp.mlp.0.bias
1394
+ dtype: F32
1395
+ shape: [3072]
1396
+ sha256: 0bbfd60a255e557d0ce1d159315a488713cfc2c9167b6af0931594cd45302c75
1397
+ - name: blocks.9.mlp.mlp.0.weight
1398
+ dtype: F32
1399
+ shape: [3072, 768]
1400
+ sha256: 2741f90b0bbe41168190009cdf429e09b1183ec7e860a1de555d2cacd4829670
1401
+ - name: blocks.9.mlp.mlp.2.bias
1402
+ dtype: F32
1403
+ shape: [768]
1404
+ sha256: d0526ead3f3cffa614006bce248f2b6b0ff4f3e1ea15c266634aab89cee394dc
1405
+ - name: blocks.9.mlp.mlp.2.weight
1406
+ dtype: F32
1407
+ shape: [768, 3072]
1408
+ sha256: 374fdd0b6b8c6fcc945dd535dd3298d176de572c130810b3f61ec03e350195b5
1409
+ - name: input_layer.bias
1410
+ dtype: F32
1411
+ shape: [768]
1412
+ sha256: ab438cc9e2483ea9a5b7dad50fc4f1b1b887176b99536de46207d52a48c05ba9
1413
+ - name: input_layer.weight
1414
+ dtype: F32
1415
+ shape: [768, 8]
1416
+ sha256: c7dda2e9d4f24fd7bc06a4f86d604560c40d12f69a5054199a5d87e73301080a
1417
+ - name: offset_perturbation
1418
+ dtype: F32
1419
+ shape: [32, 3]
1420
+ sha256: 7561371db21fcefa3962cc47fdf33102494321f1a6882cea543ece5603179fe9
1421
+ - name: out_layer.bias
1422
+ dtype: F32
1423
+ shape: [448]
1424
+ sha256: b5bf5e3ba86c98ab93001b87b2a64f7fc98f774202e7a545c20abc7eb2e6492a
1425
+ - name: out_layer.weight
1426
+ dtype: F32
1427
+ shape: [448, 768]
1428
+ sha256: 0038a57fc59333aeed1b024e028de6146f705d9d0ac7d6c85400f2254d94b560
checkpoints/conversion_metadata/slat_decoder_gs_4.yaml ADDED
@@ -0,0 +1,1428 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ format_version: 1
2
+ source_file: weights/sam-3d-objects/checkpoints/slat_decoder_gs_4.ckpt
3
+ source_sha256: 731a0eceaa47945b52aa27f650d695b2aea9cc70945751e5609e5cb5b49f0186
4
+ safetensors_file: slat_decoder_gs_4.safetensors
5
+ safetensors_files: {}
6
+ created_at_unix: 1777985861
7
+ tensor_count: 101
8
+ total_tensor_bytes: 340270352
9
+ metadata:
10
+ offset_perturbation:
11
+ $type: tensor_ref
12
+ dtype: F32
13
+ storage_key: '0'
14
+ offset_elems: 0
15
+ shape:
16
+ - 4
17
+ - 3
18
+ stride:
19
+ - 3
20
+ - 1
21
+ input_layer.weight:
22
+ $type: tensor_ref
23
+ dtype: F32
24
+ storage_key: '1'
25
+ offset_elems: 0
26
+ shape:
27
+ - 768
28
+ - 8
29
+ stride:
30
+ - 8
31
+ - 1
32
+ input_layer.bias:
33
+ $type: tensor_ref
34
+ dtype: F32
35
+ storage_key: '2'
36
+ offset_elems: 0
37
+ shape:
38
+ - 768
39
+ stride:
40
+ - 1
41
+ blocks.0.attn.to_qkv.weight:
42
+ $type: tensor_ref
43
+ dtype: F16
44
+ storage_key: '3'
45
+ offset_elems: 0
46
+ shape:
47
+ - 2304
48
+ - 768
49
+ stride:
50
+ - 768
51
+ - 1
52
+ blocks.0.attn.to_qkv.bias:
53
+ $type: tensor_ref
54
+ dtype: F16
55
+ storage_key: '4'
56
+ offset_elems: 0
57
+ shape:
58
+ - 2304
59
+ stride:
60
+ - 1
61
+ blocks.0.attn.to_out.weight:
62
+ $type: tensor_ref
63
+ dtype: F16
64
+ storage_key: '5'
65
+ offset_elems: 0
66
+ shape:
67
+ - 768
68
+ - 768
69
+ stride:
70
+ - 768
71
+ - 1
72
+ blocks.0.attn.to_out.bias:
73
+ $type: tensor_ref
74
+ dtype: F16
75
+ storage_key: '6'
76
+ offset_elems: 0
77
+ shape:
78
+ - 768
79
+ stride:
80
+ - 1
81
+ blocks.0.mlp.mlp.0.weight:
82
+ $type: tensor_ref
83
+ dtype: F16
84
+ storage_key: '7'
85
+ offset_elems: 0
86
+ shape:
87
+ - 3072
88
+ - 768
89
+ stride:
90
+ - 768
91
+ - 1
92
+ blocks.0.mlp.mlp.0.bias:
93
+ $type: tensor_ref
94
+ dtype: F16
95
+ storage_key: '8'
96
+ offset_elems: 0
97
+ shape:
98
+ - 3072
99
+ stride:
100
+ - 1
101
+ blocks.0.mlp.mlp.2.weight:
102
+ $type: tensor_ref
103
+ dtype: F16
104
+ storage_key: '9'
105
+ offset_elems: 0
106
+ shape:
107
+ - 768
108
+ - 3072
109
+ stride:
110
+ - 3072
111
+ - 1
112
+ blocks.0.mlp.mlp.2.bias:
113
+ $type: tensor_ref
114
+ dtype: F16
115
+ storage_key: '10'
116
+ offset_elems: 0
117
+ shape:
118
+ - 768
119
+ stride:
120
+ - 1
121
+ blocks.1.attn.to_qkv.weight:
122
+ $type: tensor_ref
123
+ dtype: F16
124
+ storage_key: '11'
125
+ offset_elems: 0
126
+ shape:
127
+ - 2304
128
+ - 768
129
+ stride:
130
+ - 768
131
+ - 1
132
+ blocks.1.attn.to_qkv.bias:
133
+ $type: tensor_ref
134
+ dtype: F16
135
+ storage_key: '12'
136
+ offset_elems: 0
137
+ shape:
138
+ - 2304
139
+ stride:
140
+ - 1
141
+ blocks.1.attn.to_out.weight:
142
+ $type: tensor_ref
143
+ dtype: F16
144
+ storage_key: '13'
145
+ offset_elems: 0
146
+ shape:
147
+ - 768
148
+ - 768
149
+ stride:
150
+ - 768
151
+ - 1
152
+ blocks.1.attn.to_out.bias:
153
+ $type: tensor_ref
154
+ dtype: F16
155
+ storage_key: '14'
156
+ offset_elems: 0
157
+ shape:
158
+ - 768
159
+ stride:
160
+ - 1
161
+ blocks.1.mlp.mlp.0.weight:
162
+ $type: tensor_ref
163
+ dtype: F16
164
+ storage_key: '15'
165
+ offset_elems: 0
166
+ shape:
167
+ - 3072
168
+ - 768
169
+ stride:
170
+ - 768
171
+ - 1
172
+ blocks.1.mlp.mlp.0.bias:
173
+ $type: tensor_ref
174
+ dtype: F16
175
+ storage_key: '16'
176
+ offset_elems: 0
177
+ shape:
178
+ - 3072
179
+ stride:
180
+ - 1
181
+ blocks.1.mlp.mlp.2.weight:
182
+ $type: tensor_ref
183
+ dtype: F16
184
+ storage_key: '17'
185
+ offset_elems: 0
186
+ shape:
187
+ - 768
188
+ - 3072
189
+ stride:
190
+ - 3072
191
+ - 1
192
+ blocks.1.mlp.mlp.2.bias:
193
+ $type: tensor_ref
194
+ dtype: F16
195
+ storage_key: '18'
196
+ offset_elems: 0
197
+ shape:
198
+ - 768
199
+ stride:
200
+ - 1
201
+ blocks.2.attn.to_qkv.weight:
202
+ $type: tensor_ref
203
+ dtype: F16
204
+ storage_key: '19'
205
+ offset_elems: 0
206
+ shape:
207
+ - 2304
208
+ - 768
209
+ stride:
210
+ - 768
211
+ - 1
212
+ blocks.2.attn.to_qkv.bias:
213
+ $type: tensor_ref
214
+ dtype: F16
215
+ storage_key: '20'
216
+ offset_elems: 0
217
+ shape:
218
+ - 2304
219
+ stride:
220
+ - 1
221
+ blocks.2.attn.to_out.weight:
222
+ $type: tensor_ref
223
+ dtype: F16
224
+ storage_key: '21'
225
+ offset_elems: 0
226
+ shape:
227
+ - 768
228
+ - 768
229
+ stride:
230
+ - 768
231
+ - 1
232
+ blocks.2.attn.to_out.bias:
233
+ $type: tensor_ref
234
+ dtype: F16
235
+ storage_key: '22'
236
+ offset_elems: 0
237
+ shape:
238
+ - 768
239
+ stride:
240
+ - 1
241
+ blocks.2.mlp.mlp.0.weight:
242
+ $type: tensor_ref
243
+ dtype: F16
244
+ storage_key: '23'
245
+ offset_elems: 0
246
+ shape:
247
+ - 3072
248
+ - 768
249
+ stride:
250
+ - 768
251
+ - 1
252
+ blocks.2.mlp.mlp.0.bias:
253
+ $type: tensor_ref
254
+ dtype: F16
255
+ storage_key: '24'
256
+ offset_elems: 0
257
+ shape:
258
+ - 3072
259
+ stride:
260
+ - 1
261
+ blocks.2.mlp.mlp.2.weight:
262
+ $type: tensor_ref
263
+ dtype: F16
264
+ storage_key: '25'
265
+ offset_elems: 0
266
+ shape:
267
+ - 768
268
+ - 3072
269
+ stride:
270
+ - 3072
271
+ - 1
272
+ blocks.2.mlp.mlp.2.bias:
273
+ $type: tensor_ref
274
+ dtype: F16
275
+ storage_key: '26'
276
+ offset_elems: 0
277
+ shape:
278
+ - 768
279
+ stride:
280
+ - 1
281
+ blocks.3.attn.to_qkv.weight:
282
+ $type: tensor_ref
283
+ dtype: F16
284
+ storage_key: '27'
285
+ offset_elems: 0
286
+ shape:
287
+ - 2304
288
+ - 768
289
+ stride:
290
+ - 768
291
+ - 1
292
+ blocks.3.attn.to_qkv.bias:
293
+ $type: tensor_ref
294
+ dtype: F16
295
+ storage_key: '28'
296
+ offset_elems: 0
297
+ shape:
298
+ - 2304
299
+ stride:
300
+ - 1
301
+ blocks.3.attn.to_out.weight:
302
+ $type: tensor_ref
303
+ dtype: F16
304
+ storage_key: '29'
305
+ offset_elems: 0
306
+ shape:
307
+ - 768
308
+ - 768
309
+ stride:
310
+ - 768
311
+ - 1
312
+ blocks.3.attn.to_out.bias:
313
+ $type: tensor_ref
314
+ dtype: F16
315
+ storage_key: '30'
316
+ offset_elems: 0
317
+ shape:
318
+ - 768
319
+ stride:
320
+ - 1
321
+ blocks.3.mlp.mlp.0.weight:
322
+ $type: tensor_ref
323
+ dtype: F16
324
+ storage_key: '31'
325
+ offset_elems: 0
326
+ shape:
327
+ - 3072
328
+ - 768
329
+ stride:
330
+ - 768
331
+ - 1
332
+ blocks.3.mlp.mlp.0.bias:
333
+ $type: tensor_ref
334
+ dtype: F16
335
+ storage_key: '32'
336
+ offset_elems: 0
337
+ shape:
338
+ - 3072
339
+ stride:
340
+ - 1
341
+ blocks.3.mlp.mlp.2.weight:
342
+ $type: tensor_ref
343
+ dtype: F16
344
+ storage_key: '33'
345
+ offset_elems: 0
346
+ shape:
347
+ - 768
348
+ - 3072
349
+ stride:
350
+ - 3072
351
+ - 1
352
+ blocks.3.mlp.mlp.2.bias:
353
+ $type: tensor_ref
354
+ dtype: F16
355
+ storage_key: '34'
356
+ offset_elems: 0
357
+ shape:
358
+ - 768
359
+ stride:
360
+ - 1
361
+ blocks.4.attn.to_qkv.weight:
362
+ $type: tensor_ref
363
+ dtype: F16
364
+ storage_key: '35'
365
+ offset_elems: 0
366
+ shape:
367
+ - 2304
368
+ - 768
369
+ stride:
370
+ - 768
371
+ - 1
372
+ blocks.4.attn.to_qkv.bias:
373
+ $type: tensor_ref
374
+ dtype: F16
375
+ storage_key: '36'
376
+ offset_elems: 0
377
+ shape:
378
+ - 2304
379
+ stride:
380
+ - 1
381
+ blocks.4.attn.to_out.weight:
382
+ $type: tensor_ref
383
+ dtype: F16
384
+ storage_key: '37'
385
+ offset_elems: 0
386
+ shape:
387
+ - 768
388
+ - 768
389
+ stride:
390
+ - 768
391
+ - 1
392
+ blocks.4.attn.to_out.bias:
393
+ $type: tensor_ref
394
+ dtype: F16
395
+ storage_key: '38'
396
+ offset_elems: 0
397
+ shape:
398
+ - 768
399
+ stride:
400
+ - 1
401
+ blocks.4.mlp.mlp.0.weight:
402
+ $type: tensor_ref
403
+ dtype: F16
404
+ storage_key: '39'
405
+ offset_elems: 0
406
+ shape:
407
+ - 3072
408
+ - 768
409
+ stride:
410
+ - 768
411
+ - 1
412
+ blocks.4.mlp.mlp.0.bias:
413
+ $type: tensor_ref
414
+ dtype: F16
415
+ storage_key: '40'
416
+ offset_elems: 0
417
+ shape:
418
+ - 3072
419
+ stride:
420
+ - 1
421
+ blocks.4.mlp.mlp.2.weight:
422
+ $type: tensor_ref
423
+ dtype: F16
424
+ storage_key: '41'
425
+ offset_elems: 0
426
+ shape:
427
+ - 768
428
+ - 3072
429
+ stride:
430
+ - 3072
431
+ - 1
432
+ blocks.4.mlp.mlp.2.bias:
433
+ $type: tensor_ref
434
+ dtype: F16
435
+ storage_key: '42'
436
+ offset_elems: 0
437
+ shape:
438
+ - 768
439
+ stride:
440
+ - 1
441
+ blocks.5.attn.to_qkv.weight:
442
+ $type: tensor_ref
443
+ dtype: F16
444
+ storage_key: '43'
445
+ offset_elems: 0
446
+ shape:
447
+ - 2304
448
+ - 768
449
+ stride:
450
+ - 768
451
+ - 1
452
+ blocks.5.attn.to_qkv.bias:
453
+ $type: tensor_ref
454
+ dtype: F16
455
+ storage_key: '44'
456
+ offset_elems: 0
457
+ shape:
458
+ - 2304
459
+ stride:
460
+ - 1
461
+ blocks.5.attn.to_out.weight:
462
+ $type: tensor_ref
463
+ dtype: F16
464
+ storage_key: '45'
465
+ offset_elems: 0
466
+ shape:
467
+ - 768
468
+ - 768
469
+ stride:
470
+ - 768
471
+ - 1
472
+ blocks.5.attn.to_out.bias:
473
+ $type: tensor_ref
474
+ dtype: F16
475
+ storage_key: '46'
476
+ offset_elems: 0
477
+ shape:
478
+ - 768
479
+ stride:
480
+ - 1
481
+ blocks.5.mlp.mlp.0.weight:
482
+ $type: tensor_ref
483
+ dtype: F16
484
+ storage_key: '47'
485
+ offset_elems: 0
486
+ shape:
487
+ - 3072
488
+ - 768
489
+ stride:
490
+ - 768
491
+ - 1
492
+ blocks.5.mlp.mlp.0.bias:
493
+ $type: tensor_ref
494
+ dtype: F16
495
+ storage_key: '48'
496
+ offset_elems: 0
497
+ shape:
498
+ - 3072
499
+ stride:
500
+ - 1
501
+ blocks.5.mlp.mlp.2.weight:
502
+ $type: tensor_ref
503
+ dtype: F16
504
+ storage_key: '49'
505
+ offset_elems: 0
506
+ shape:
507
+ - 768
508
+ - 3072
509
+ stride:
510
+ - 3072
511
+ - 1
512
+ blocks.5.mlp.mlp.2.bias:
513
+ $type: tensor_ref
514
+ dtype: F16
515
+ storage_key: '50'
516
+ offset_elems: 0
517
+ shape:
518
+ - 768
519
+ stride:
520
+ - 1
521
+ blocks.6.attn.to_qkv.weight:
522
+ $type: tensor_ref
523
+ dtype: F16
524
+ storage_key: '51'
525
+ offset_elems: 0
526
+ shape:
527
+ - 2304
528
+ - 768
529
+ stride:
530
+ - 768
531
+ - 1
532
+ blocks.6.attn.to_qkv.bias:
533
+ $type: tensor_ref
534
+ dtype: F16
535
+ storage_key: '52'
536
+ offset_elems: 0
537
+ shape:
538
+ - 2304
539
+ stride:
540
+ - 1
541
+ blocks.6.attn.to_out.weight:
542
+ $type: tensor_ref
543
+ dtype: F16
544
+ storage_key: '53'
545
+ offset_elems: 0
546
+ shape:
547
+ - 768
548
+ - 768
549
+ stride:
550
+ - 768
551
+ - 1
552
+ blocks.6.attn.to_out.bias:
553
+ $type: tensor_ref
554
+ dtype: F16
555
+ storage_key: '54'
556
+ offset_elems: 0
557
+ shape:
558
+ - 768
559
+ stride:
560
+ - 1
561
+ blocks.6.mlp.mlp.0.weight:
562
+ $type: tensor_ref
563
+ dtype: F16
564
+ storage_key: '55'
565
+ offset_elems: 0
566
+ shape:
567
+ - 3072
568
+ - 768
569
+ stride:
570
+ - 768
571
+ - 1
572
+ blocks.6.mlp.mlp.0.bias:
573
+ $type: tensor_ref
574
+ dtype: F16
575
+ storage_key: '56'
576
+ offset_elems: 0
577
+ shape:
578
+ - 3072
579
+ stride:
580
+ - 1
581
+ blocks.6.mlp.mlp.2.weight:
582
+ $type: tensor_ref
583
+ dtype: F16
584
+ storage_key: '57'
585
+ offset_elems: 0
586
+ shape:
587
+ - 768
588
+ - 3072
589
+ stride:
590
+ - 3072
591
+ - 1
592
+ blocks.6.mlp.mlp.2.bias:
593
+ $type: tensor_ref
594
+ dtype: F16
595
+ storage_key: '58'
596
+ offset_elems: 0
597
+ shape:
598
+ - 768
599
+ stride:
600
+ - 1
601
+ blocks.7.attn.to_qkv.weight:
602
+ $type: tensor_ref
603
+ dtype: F16
604
+ storage_key: '59'
605
+ offset_elems: 0
606
+ shape:
607
+ - 2304
608
+ - 768
609
+ stride:
610
+ - 768
611
+ - 1
612
+ blocks.7.attn.to_qkv.bias:
613
+ $type: tensor_ref
614
+ dtype: F16
615
+ storage_key: '60'
616
+ offset_elems: 0
617
+ shape:
618
+ - 2304
619
+ stride:
620
+ - 1
621
+ blocks.7.attn.to_out.weight:
622
+ $type: tensor_ref
623
+ dtype: F16
624
+ storage_key: '61'
625
+ offset_elems: 0
626
+ shape:
627
+ - 768
628
+ - 768
629
+ stride:
630
+ - 768
631
+ - 1
632
+ blocks.7.attn.to_out.bias:
633
+ $type: tensor_ref
634
+ dtype: F16
635
+ storage_key: '62'
636
+ offset_elems: 0
637
+ shape:
638
+ - 768
639
+ stride:
640
+ - 1
641
+ blocks.7.mlp.mlp.0.weight:
642
+ $type: tensor_ref
643
+ dtype: F16
644
+ storage_key: '63'
645
+ offset_elems: 0
646
+ shape:
647
+ - 3072
648
+ - 768
649
+ stride:
650
+ - 768
651
+ - 1
652
+ blocks.7.mlp.mlp.0.bias:
653
+ $type: tensor_ref
654
+ dtype: F16
655
+ storage_key: '64'
656
+ offset_elems: 0
657
+ shape:
658
+ - 3072
659
+ stride:
660
+ - 1
661
+ blocks.7.mlp.mlp.2.weight:
662
+ $type: tensor_ref
663
+ dtype: F16
664
+ storage_key: '65'
665
+ offset_elems: 0
666
+ shape:
667
+ - 768
668
+ - 3072
669
+ stride:
670
+ - 3072
671
+ - 1
672
+ blocks.7.mlp.mlp.2.bias:
673
+ $type: tensor_ref
674
+ dtype: F16
675
+ storage_key: '66'
676
+ offset_elems: 0
677
+ shape:
678
+ - 768
679
+ stride:
680
+ - 1
681
+ blocks.8.attn.to_qkv.weight:
682
+ $type: tensor_ref
683
+ dtype: F16
684
+ storage_key: '67'
685
+ offset_elems: 0
686
+ shape:
687
+ - 2304
688
+ - 768
689
+ stride:
690
+ - 768
691
+ - 1
692
+ blocks.8.attn.to_qkv.bias:
693
+ $type: tensor_ref
694
+ dtype: F16
695
+ storage_key: '68'
696
+ offset_elems: 0
697
+ shape:
698
+ - 2304
699
+ stride:
700
+ - 1
701
+ blocks.8.attn.to_out.weight:
702
+ $type: tensor_ref
703
+ dtype: F16
704
+ storage_key: '69'
705
+ offset_elems: 0
706
+ shape:
707
+ - 768
708
+ - 768
709
+ stride:
710
+ - 768
711
+ - 1
712
+ blocks.8.attn.to_out.bias:
713
+ $type: tensor_ref
714
+ dtype: F16
715
+ storage_key: '70'
716
+ offset_elems: 0
717
+ shape:
718
+ - 768
719
+ stride:
720
+ - 1
721
+ blocks.8.mlp.mlp.0.weight:
722
+ $type: tensor_ref
723
+ dtype: F16
724
+ storage_key: '71'
725
+ offset_elems: 0
726
+ shape:
727
+ - 3072
728
+ - 768
729
+ stride:
730
+ - 768
731
+ - 1
732
+ blocks.8.mlp.mlp.0.bias:
733
+ $type: tensor_ref
734
+ dtype: F16
735
+ storage_key: '72'
736
+ offset_elems: 0
737
+ shape:
738
+ - 3072
739
+ stride:
740
+ - 1
741
+ blocks.8.mlp.mlp.2.weight:
742
+ $type: tensor_ref
743
+ dtype: F16
744
+ storage_key: '73'
745
+ offset_elems: 0
746
+ shape:
747
+ - 768
748
+ - 3072
749
+ stride:
750
+ - 3072
751
+ - 1
752
+ blocks.8.mlp.mlp.2.bias:
753
+ $type: tensor_ref
754
+ dtype: F16
755
+ storage_key: '74'
756
+ offset_elems: 0
757
+ shape:
758
+ - 768
759
+ stride:
760
+ - 1
761
+ blocks.9.attn.to_qkv.weight:
762
+ $type: tensor_ref
763
+ dtype: F16
764
+ storage_key: '75'
765
+ offset_elems: 0
766
+ shape:
767
+ - 2304
768
+ - 768
769
+ stride:
770
+ - 768
771
+ - 1
772
+ blocks.9.attn.to_qkv.bias:
773
+ $type: tensor_ref
774
+ dtype: F16
775
+ storage_key: '76'
776
+ offset_elems: 0
777
+ shape:
778
+ - 2304
779
+ stride:
780
+ - 1
781
+ blocks.9.attn.to_out.weight:
782
+ $type: tensor_ref
783
+ dtype: F16
784
+ storage_key: '77'
785
+ offset_elems: 0
786
+ shape:
787
+ - 768
788
+ - 768
789
+ stride:
790
+ - 768
791
+ - 1
792
+ blocks.9.attn.to_out.bias:
793
+ $type: tensor_ref
794
+ dtype: F16
795
+ storage_key: '78'
796
+ offset_elems: 0
797
+ shape:
798
+ - 768
799
+ stride:
800
+ - 1
801
+ blocks.9.mlp.mlp.0.weight:
802
+ $type: tensor_ref
803
+ dtype: F16
804
+ storage_key: '79'
805
+ offset_elems: 0
806
+ shape:
807
+ - 3072
808
+ - 768
809
+ stride:
810
+ - 768
811
+ - 1
812
+ blocks.9.mlp.mlp.0.bias:
813
+ $type: tensor_ref
814
+ dtype: F16
815
+ storage_key: '80'
816
+ offset_elems: 0
817
+ shape:
818
+ - 3072
819
+ stride:
820
+ - 1
821
+ blocks.9.mlp.mlp.2.weight:
822
+ $type: tensor_ref
823
+ dtype: F16
824
+ storage_key: '81'
825
+ offset_elems: 0
826
+ shape:
827
+ - 768
828
+ - 3072
829
+ stride:
830
+ - 3072
831
+ - 1
832
+ blocks.9.mlp.mlp.2.bias:
833
+ $type: tensor_ref
834
+ dtype: F16
835
+ storage_key: '82'
836
+ offset_elems: 0
837
+ shape:
838
+ - 768
839
+ stride:
840
+ - 1
841
+ blocks.10.attn.to_qkv.weight:
842
+ $type: tensor_ref
843
+ dtype: F16
844
+ storage_key: '83'
845
+ offset_elems: 0
846
+ shape:
847
+ - 2304
848
+ - 768
849
+ stride:
850
+ - 768
851
+ - 1
852
+ blocks.10.attn.to_qkv.bias:
853
+ $type: tensor_ref
854
+ dtype: F16
855
+ storage_key: '84'
856
+ offset_elems: 0
857
+ shape:
858
+ - 2304
859
+ stride:
860
+ - 1
861
+ blocks.10.attn.to_out.weight:
862
+ $type: tensor_ref
863
+ dtype: F16
864
+ storage_key: '85'
865
+ offset_elems: 0
866
+ shape:
867
+ - 768
868
+ - 768
869
+ stride:
870
+ - 768
871
+ - 1
872
+ blocks.10.attn.to_out.bias:
873
+ $type: tensor_ref
874
+ dtype: F16
875
+ storage_key: '86'
876
+ offset_elems: 0
877
+ shape:
878
+ - 768
879
+ stride:
880
+ - 1
881
+ blocks.10.mlp.mlp.0.weight:
882
+ $type: tensor_ref
883
+ dtype: F16
884
+ storage_key: '87'
885
+ offset_elems: 0
886
+ shape:
887
+ - 3072
888
+ - 768
889
+ stride:
890
+ - 768
891
+ - 1
892
+ blocks.10.mlp.mlp.0.bias:
893
+ $type: tensor_ref
894
+ dtype: F16
895
+ storage_key: '88'
896
+ offset_elems: 0
897
+ shape:
898
+ - 3072
899
+ stride:
900
+ - 1
901
+ blocks.10.mlp.mlp.2.weight:
902
+ $type: tensor_ref
903
+ dtype: F16
904
+ storage_key: '89'
905
+ offset_elems: 0
906
+ shape:
907
+ - 768
908
+ - 3072
909
+ stride:
910
+ - 3072
911
+ - 1
912
+ blocks.10.mlp.mlp.2.bias:
913
+ $type: tensor_ref
914
+ dtype: F16
915
+ storage_key: '90'
916
+ offset_elems: 0
917
+ shape:
918
+ - 768
919
+ stride:
920
+ - 1
921
+ blocks.11.attn.to_qkv.weight:
922
+ $type: tensor_ref
923
+ dtype: F16
924
+ storage_key: '91'
925
+ offset_elems: 0
926
+ shape:
927
+ - 2304
928
+ - 768
929
+ stride:
930
+ - 768
931
+ - 1
932
+ blocks.11.attn.to_qkv.bias:
933
+ $type: tensor_ref
934
+ dtype: F16
935
+ storage_key: '92'
936
+ offset_elems: 0
937
+ shape:
938
+ - 2304
939
+ stride:
940
+ - 1
941
+ blocks.11.attn.to_out.weight:
942
+ $type: tensor_ref
943
+ dtype: F16
944
+ storage_key: '93'
945
+ offset_elems: 0
946
+ shape:
947
+ - 768
948
+ - 768
949
+ stride:
950
+ - 768
951
+ - 1
952
+ blocks.11.attn.to_out.bias:
953
+ $type: tensor_ref
954
+ dtype: F16
955
+ storage_key: '94'
956
+ offset_elems: 0
957
+ shape:
958
+ - 768
959
+ stride:
960
+ - 1
961
+ blocks.11.mlp.mlp.0.weight:
962
+ $type: tensor_ref
963
+ dtype: F16
964
+ storage_key: '95'
965
+ offset_elems: 0
966
+ shape:
967
+ - 3072
968
+ - 768
969
+ stride:
970
+ - 768
971
+ - 1
972
+ blocks.11.mlp.mlp.0.bias:
973
+ $type: tensor_ref
974
+ dtype: F16
975
+ storage_key: '96'
976
+ offset_elems: 0
977
+ shape:
978
+ - 3072
979
+ stride:
980
+ - 1
981
+ blocks.11.mlp.mlp.2.weight:
982
+ $type: tensor_ref
983
+ dtype: F16
984
+ storage_key: '97'
985
+ offset_elems: 0
986
+ shape:
987
+ - 768
988
+ - 3072
989
+ stride:
990
+ - 3072
991
+ - 1
992
+ blocks.11.mlp.mlp.2.bias:
993
+ $type: tensor_ref
994
+ dtype: F16
995
+ storage_key: '98'
996
+ offset_elems: 0
997
+ shape:
998
+ - 768
999
+ stride:
1000
+ - 1
1001
+ out_layer.weight:
1002
+ $type: tensor_ref
1003
+ dtype: F32
1004
+ storage_key: '99'
1005
+ offset_elems: 0
1006
+ shape:
1007
+ - 56
1008
+ - 768
1009
+ stride:
1010
+ - 768
1011
+ - 1
1012
+ out_layer.bias:
1013
+ $type: tensor_ref
1014
+ dtype: F32
1015
+ storage_key: '100'
1016
+ offset_elems: 0
1017
+ shape:
1018
+ - 56
1019
+ stride:
1020
+ - 1
1021
+ security:
1022
+ objects: []
1023
+ calls: []
1024
+ tensors:
1025
+ - name: blocks.0.attn.to_out.bias
1026
+ dtype: F32
1027
+ shape: [768]
1028
+ sha256: c361769a368b0f6b786dfab74c7a09bf91e022bc1d0f78a7ecfcd4b169ead136
1029
+ - name: blocks.0.attn.to_out.weight
1030
+ dtype: F32
1031
+ shape: [768, 768]
1032
+ sha256: 21966b9472b7450bd3d02286be150767781b9f6454afd4a71895add45c76f13b
1033
+ - name: blocks.0.attn.to_qkv.bias
1034
+ dtype: F32
1035
+ shape: [2304]
1036
+ sha256: 97ab278eacfb25c444ab4db1cce4bff2ab2eebd7f851cb4b7de6a6893fc62850
1037
+ - name: blocks.0.attn.to_qkv.weight
1038
+ dtype: F32
1039
+ shape: [2304, 768]
1040
+ sha256: 48d7fec929a6195e4a822c9ec222727325d6195651c1186f8bb0815a15452e72
1041
+ - name: blocks.0.mlp.mlp.0.bias
1042
+ dtype: F32
1043
+ shape: [3072]
1044
+ sha256: b9dc315e6529c6e94e2d64f35782a190deb8a15efea566c495f539aac18de38e
1045
+ - name: blocks.0.mlp.mlp.0.weight
1046
+ dtype: F32
1047
+ shape: [3072, 768]
1048
+ sha256: e66b27b9946e4109d566cc1566bce0896d4537a1fb63b0942edbb90a38b3354a
1049
+ - name: blocks.0.mlp.mlp.2.bias
1050
+ dtype: F32
1051
+ shape: [768]
1052
+ sha256: c5962f2a81b43c2021289549dcebfc199446c8babb2d1917a07558857394d077
1053
+ - name: blocks.0.mlp.mlp.2.weight
1054
+ dtype: F32
1055
+ shape: [768, 3072]
1056
+ sha256: b25aa4cd7ef209ee8f5478aa943f67b00d3a13abc310a0a1420ae236bfe61d11
1057
+ - name: blocks.1.attn.to_out.bias
1058
+ dtype: F32
1059
+ shape: [768]
1060
+ sha256: 1703062a1049a81a47aee31f6afaa7dc976106485b0ccc385ad0c49d2a6bbc92
1061
+ - name: blocks.1.attn.to_out.weight
1062
+ dtype: F32
1063
+ shape: [768, 768]
1064
+ sha256: 3a5e71b2451130f30c2b90ae995b1277b3abb29d33ccbd1a9759a9f513003c63
1065
+ - name: blocks.1.attn.to_qkv.bias
1066
+ dtype: F32
1067
+ shape: [2304]
1068
+ sha256: 9c153e5940d0de4cbd7f150cb7dd4f29407a7adf12db1fdefff97f79f6c72bcc
1069
+ - name: blocks.1.attn.to_qkv.weight
1070
+ dtype: F32
1071
+ shape: [2304, 768]
1072
+ sha256: a9296ad4497400614a7dbcbdf7d617a6f887ab02a265cc8e264174d09239e061
1073
+ - name: blocks.1.mlp.mlp.0.bias
1074
+ dtype: F32
1075
+ shape: [3072]
1076
+ sha256: f1c800868fa2314f43acff6f70512053266fd78d5592bb60c24e2d19175d1105
1077
+ - name: blocks.1.mlp.mlp.0.weight
1078
+ dtype: F32
1079
+ shape: [3072, 768]
1080
+ sha256: c34a6039c66fb570ec0e964245606e98fe7919f1e217cc659cf5db734affd293
1081
+ - name: blocks.1.mlp.mlp.2.bias
1082
+ dtype: F32
1083
+ shape: [768]
1084
+ sha256: 8db35efc0665403b6df4498766b8ee04778704bb803653fc68823d5e4a3daf23
1085
+ - name: blocks.1.mlp.mlp.2.weight
1086
+ dtype: F32
1087
+ shape: [768, 3072]
1088
+ sha256: 276cc495a51c9a6357f2867e0689535fe216fcc5eef20785b9245d0105b5ea11
1089
+ - name: blocks.10.attn.to_out.bias
1090
+ dtype: F32
1091
+ shape: [768]
1092
+ sha256: 6258f49adccd093555bb73bf699568ff2bfc7e61afd84d583ddd459bb78bad36
1093
+ - name: blocks.10.attn.to_out.weight
1094
+ dtype: F32
1095
+ shape: [768, 768]
1096
+ sha256: e643a19c70e879fdef9a1924e183ec99ebd47bb5bbed011d53d62d87f8997f77
1097
+ - name: blocks.10.attn.to_qkv.bias
1098
+ dtype: F32
1099
+ shape: [2304]
1100
+ sha256: 08b807c79f1572f029ffa35e3b3ed31fe2493953f22b3ae1401f0b6594eff9af
1101
+ - name: blocks.10.attn.to_qkv.weight
1102
+ dtype: F32
1103
+ shape: [2304, 768]
1104
+ sha256: 01321f38934265f697c63650e903b783444f72a39fbe1972728015d48a4f7aec
1105
+ - name: blocks.10.mlp.mlp.0.bias
1106
+ dtype: F32
1107
+ shape: [3072]
1108
+ sha256: 461b809f9826856c98b43953952b5e71d4db03f08cde75e8bc14858bd72c33f2
1109
+ - name: blocks.10.mlp.mlp.0.weight
1110
+ dtype: F32
1111
+ shape: [3072, 768]
1112
+ sha256: 9d94c831a2bf6d00e0e421adbbebaa06d236d731b6a2389cfed0eef70beff7a9
1113
+ - name: blocks.10.mlp.mlp.2.bias
1114
+ dtype: F32
1115
+ shape: [768]
1116
+ sha256: 81ca9442953f2b4a5ab9f126c715a3ed122ab9eb1602727e8b7e341fb2d440b1
1117
+ - name: blocks.10.mlp.mlp.2.weight
1118
+ dtype: F32
1119
+ shape: [768, 3072]
1120
+ sha256: c877c513f4a0d4b5472067fc1d817049e3bc7104ca68a4feed3dd413e3cadd34
1121
+ - name: blocks.11.attn.to_out.bias
1122
+ dtype: F32
1123
+ shape: [768]
1124
+ sha256: 3733fdc7596a4c7ce42bb8c84fed24d163f4476123e42e5d25b9012e2e117cdd
1125
+ - name: blocks.11.attn.to_out.weight
1126
+ dtype: F32
1127
+ shape: [768, 768]
1128
+ sha256: bc838b835d0b3137f4ed2de52614d6f767c0c773b5ca74f2ef48e526fa638e1c
1129
+ - name: blocks.11.attn.to_qkv.bias
1130
+ dtype: F32
1131
+ shape: [2304]
1132
+ sha256: 255c502955163ef6904227ec95214ecaa6d64dfb7f1ee927c97a022c52b40e57
1133
+ - name: blocks.11.attn.to_qkv.weight
1134
+ dtype: F32
1135
+ shape: [2304, 768]
1136
+ sha256: 6255e19f8f28f8e5ebd810c49841a8a4a1110cee8da95fb06f2f663d4c984d1d
1137
+ - name: blocks.11.mlp.mlp.0.bias
1138
+ dtype: F32
1139
+ shape: [3072]
1140
+ sha256: 415f83ecb06a442efd5ac38ac4de4fb105ea1169a7b498d72bc1a4f9c8ab9224
1141
+ - name: blocks.11.mlp.mlp.0.weight
1142
+ dtype: F32
1143
+ shape: [3072, 768]
1144
+ sha256: 42797ce7ca75f315b987ce361e0a4f2a30ce0354fb34f7c9274a9a021fac53a6
1145
+ - name: blocks.11.mlp.mlp.2.bias
1146
+ dtype: F32
1147
+ shape: [768]
1148
+ sha256: 61989bb9103f2055a6b9e82ba04fd057172b7873bb76be9611fef1127daf9530
1149
+ - name: blocks.11.mlp.mlp.2.weight
1150
+ dtype: F32
1151
+ shape: [768, 3072]
1152
+ sha256: 033fc161a9272ab0d5cf4f2c2be7ab5b38126bcde2dc323d711618c70e28bf1a
1153
+ - name: blocks.2.attn.to_out.bias
1154
+ dtype: F32
1155
+ shape: [768]
1156
+ sha256: 20dd78aedf93be0a5c693db0000dd149ac49436e0e87ab25ff2d1096208281b6
1157
+ - name: blocks.2.attn.to_out.weight
1158
+ dtype: F32
1159
+ shape: [768, 768]
1160
+ sha256: c1a1e90bce83f8715ab5ce63915730cc3d0a162b15fe80fe623d6476bb52c65f
1161
+ - name: blocks.2.attn.to_qkv.bias
1162
+ dtype: F32
1163
+ shape: [2304]
1164
+ sha256: 1053c0038761a260cc9df74247096c64bdcd52a5f8f3af267ee3b5735f9efa67
1165
+ - name: blocks.2.attn.to_qkv.weight
1166
+ dtype: F32
1167
+ shape: [2304, 768]
1168
+ sha256: 79771ae52de6ccd2d74387c9f3f6e094b09b83e90a919586e382061145b6060c
1169
+ - name: blocks.2.mlp.mlp.0.bias
1170
+ dtype: F32
1171
+ shape: [3072]
1172
+ sha256: 11fa4cfdf76d6c8f859a287fd5cb05a83d447f6d6beb97913a742ced0bd5b815
1173
+ - name: blocks.2.mlp.mlp.0.weight
1174
+ dtype: F32
1175
+ shape: [3072, 768]
1176
+ sha256: 7eecec79e7e7f78290b47ccf7967bd7fa53d5b285f2c555c4931851168051683
1177
+ - name: blocks.2.mlp.mlp.2.bias
1178
+ dtype: F32
1179
+ shape: [768]
1180
+ sha256: 1d5c8aecf77b2cc6eb8647648b50ef2617e3c1084c2a5db85962b1683f073d40
1181
+ - name: blocks.2.mlp.mlp.2.weight
1182
+ dtype: F32
1183
+ shape: [768, 3072]
1184
+ sha256: 925b494856f05572404f515aa9e9c0e92114e438d5415248c11b6f177f78fb65
1185
+ - name: blocks.3.attn.to_out.bias
1186
+ dtype: F32
1187
+ shape: [768]
1188
+ sha256: 269de532b9fe16fa00fd5f4cadc86885be10e0b6b50f9408108150a0d746853b
1189
+ - name: blocks.3.attn.to_out.weight
1190
+ dtype: F32
1191
+ shape: [768, 768]
1192
+ sha256: 8ecab240bb9e870e9caa18a9c1ecc97cbf9ec04fa68645423ec99fa7fbda1b47
1193
+ - name: blocks.3.attn.to_qkv.bias
1194
+ dtype: F32
1195
+ shape: [2304]
1196
+ sha256: 0f9b9ab94dca308b89a382f4ab71d73f2149011504700855139bb0a0c71d1fe2
1197
+ - name: blocks.3.attn.to_qkv.weight
1198
+ dtype: F32
1199
+ shape: [2304, 768]
1200
+ sha256: 0c67b2ec6bb2b9285ebd4df9746a056c7526b7cb5635d73deda4bc51b35921a1
1201
+ - name: blocks.3.mlp.mlp.0.bias
1202
+ dtype: F32
1203
+ shape: [3072]
1204
+ sha256: cf85562d9a6e5cba48c112a2cf3341407ae5e20aae6a67ebe55a349903efe7c1
1205
+ - name: blocks.3.mlp.mlp.0.weight
1206
+ dtype: F32
1207
+ shape: [3072, 768]
1208
+ sha256: f9a41c4234cbbd71be49eaa72f947c84114da393e7eb06d11798283ac02efab9
1209
+ - name: blocks.3.mlp.mlp.2.bias
1210
+ dtype: F32
1211
+ shape: [768]
1212
+ sha256: 64070d3e0dddb4217e2bcca0f1a3fdc1f460fda92a4f95a2863ba59d5b9e51ef
1213
+ - name: blocks.3.mlp.mlp.2.weight
1214
+ dtype: F32
1215
+ shape: [768, 3072]
1216
+ sha256: cf96cf53383c7ea40501c33b7c81e008f0772333be30cb2ba2a0e527c6902300
1217
+ - name: blocks.4.attn.to_out.bias
1218
+ dtype: F32
1219
+ shape: [768]
1220
+ sha256: 888d399a5a22dede36fa24d14e8db1f170449702562901f714752de587259a67
1221
+ - name: blocks.4.attn.to_out.weight
1222
+ dtype: F32
1223
+ shape: [768, 768]
1224
+ sha256: c16a74c864d669578b158928d05f0cacfe0be053fa794fb94ede27592920e207
1225
+ - name: blocks.4.attn.to_qkv.bias
1226
+ dtype: F32
1227
+ shape: [2304]
1228
+ sha256: 6bbbf24704b6e6e234f9995e535f7052559a05a4653bce34421f5e1adcda5acf
1229
+ - name: blocks.4.attn.to_qkv.weight
1230
+ dtype: F32
1231
+ shape: [2304, 768]
1232
+ sha256: 60671cd78fa50e6f902fb8ffe8f43a1a46abc73e2dc453b1aa5eb722fa28def8
1233
+ - name: blocks.4.mlp.mlp.0.bias
1234
+ dtype: F32
1235
+ shape: [3072]
1236
+ sha256: eab9f4a0508499f43b1cd8745f79721d7927cb4056a299c69c93a075510fcf1c
1237
+ - name: blocks.4.mlp.mlp.0.weight
1238
+ dtype: F32
1239
+ shape: [3072, 768]
1240
+ sha256: 780cf39e9e81c415a9d5be136439537acdbaaea00b75e752343a0b9c2baba2e8
1241
+ - name: blocks.4.mlp.mlp.2.bias
1242
+ dtype: F32
1243
+ shape: [768]
1244
+ sha256: b3a0ece7c44f48a485f8688581d16cf9acb2d6b973157338ddf1ab32534cd417
1245
+ - name: blocks.4.mlp.mlp.2.weight
1246
+ dtype: F32
1247
+ shape: [768, 3072]
1248
+ sha256: f32931d1884cc185cad2db292fe2a2f089f97a112219f1dfa5c7b3df7e171b53
1249
+ - name: blocks.5.attn.to_out.bias
1250
+ dtype: F32
1251
+ shape: [768]
1252
+ sha256: 46bc11af50636f54d55395f6a621d4f1669d82752fe76b705da5ab7ab6e47394
1253
+ - name: blocks.5.attn.to_out.weight
1254
+ dtype: F32
1255
+ shape: [768, 768]
1256
+ sha256: d664b4ecd8f8f4a22d6c7e01e02de8b06b53186ee6342fa5533cbf841a890602
1257
+ - name: blocks.5.attn.to_qkv.bias
1258
+ dtype: F32
1259
+ shape: [2304]
1260
+ sha256: 9351acef01b27bf7c8a12e8e9d087c012592be09ee717c7387971899d15eae97
1261
+ - name: blocks.5.attn.to_qkv.weight
1262
+ dtype: F32
1263
+ shape: [2304, 768]
1264
+ sha256: a3a8865c2c6c740135647f7e873256d3443d191b0748bf79f3a921e95c7e0380
1265
+ - name: blocks.5.mlp.mlp.0.bias
1266
+ dtype: F32
1267
+ shape: [3072]
1268
+ sha256: fa49fe70fb0eb25c29da2ac61f717cbbdde19e8b461c6a0d7ed2391bc0f1a9df
1269
+ - name: blocks.5.mlp.mlp.0.weight
1270
+ dtype: F32
1271
+ shape: [3072, 768]
1272
+ sha256: f42117f69cc2d84393023dc80a8c6bf4868922b86f56374b4d43a55d1837d031
1273
+ - name: blocks.5.mlp.mlp.2.bias
1274
+ dtype: F32
1275
+ shape: [768]
1276
+ sha256: 58ee71ac766e6ebe459f376aba3b29b7b7c688d1bafaca38992c8ec3039c0e75
1277
+ - name: blocks.5.mlp.mlp.2.weight
1278
+ dtype: F32
1279
+ shape: [768, 3072]
1280
+ sha256: f0b8a04872c0e551503bb61ab28251290b1112fa65cff0c78658447d4ef63a01
1281
+ - name: blocks.6.attn.to_out.bias
1282
+ dtype: F32
1283
+ shape: [768]
1284
+ sha256: 0b957de0cc8aaa58230a9c58fcb2ecfaecf91be15cd09207dfb30d5352e96f0d
1285
+ - name: blocks.6.attn.to_out.weight
1286
+ dtype: F32
1287
+ shape: [768, 768]
1288
+ sha256: cb7f8fd5d1a31c17932cf44a2fb6a026eaf8ccc4c1de696951c5c38914d2b21b
1289
+ - name: blocks.6.attn.to_qkv.bias
1290
+ dtype: F32
1291
+ shape: [2304]
1292
+ sha256: f9b525182625232e019f675f57781e64461a2b51bb3c458bf735ece3b6f3ae00
1293
+ - name: blocks.6.attn.to_qkv.weight
1294
+ dtype: F32
1295
+ shape: [2304, 768]
1296
+ sha256: 2942a5fc747135d1719227c91e93dc65b7745d122b6ceb1dacd68be526c9a74f
1297
+ - name: blocks.6.mlp.mlp.0.bias
1298
+ dtype: F32
1299
+ shape: [3072]
1300
+ sha256: 465d8807bcdb65128cb86f4b929d8c976583eaf7c049fb9d348abb3c4b345d50
1301
+ - name: blocks.6.mlp.mlp.0.weight
1302
+ dtype: F32
1303
+ shape: [3072, 768]
1304
+ sha256: 398ba14c1934ed8a69c95738f66ca62e528f821fc4051089b981c7a87f56aa74
1305
+ - name: blocks.6.mlp.mlp.2.bias
1306
+ dtype: F32
1307
+ shape: [768]
1308
+ sha256: 48eb289e591e7c4bb6e6514ae0a2dd303d0042494ad6a7685169b393f4cb719a
1309
+ - name: blocks.6.mlp.mlp.2.weight
1310
+ dtype: F32
1311
+ shape: [768, 3072]
1312
+ sha256: eed0cc0d9c4a83a9a6df6f013034ce1d46e2803fa0553c67363ede28bbbaf48a
1313
+ - name: blocks.7.attn.to_out.bias
1314
+ dtype: F32
1315
+ shape: [768]
1316
+ sha256: 61723d80ef8c9cd81956a6df7db244653f7e9af75d8c7ecba79b94d888c92e4a
1317
+ - name: blocks.7.attn.to_out.weight
1318
+ dtype: F32
1319
+ shape: [768, 768]
1320
+ sha256: d3b4cf491c2a2e9962061ab2478a3f6538383f8a32db23a79994c2b360b5c9c0
1321
+ - name: blocks.7.attn.to_qkv.bias
1322
+ dtype: F32
1323
+ shape: [2304]
1324
+ sha256: 173cfda0f8a4d92db1016c447aff3b8478bfd23543f2c1ebea9312d7aefa2351
1325
+ - name: blocks.7.attn.to_qkv.weight
1326
+ dtype: F32
1327
+ shape: [2304, 768]
1328
+ sha256: 39abffeacc992bcd9e6a8c3be3716957ca6410b6d5c8d218710b7ff5abe3e692
1329
+ - name: blocks.7.mlp.mlp.0.bias
1330
+ dtype: F32
1331
+ shape: [3072]
1332
+ sha256: f46cfbc5d994519faeb5d11e03e57d8cc7a427f67d2b306883ca17c0f4e7be9a
1333
+ - name: blocks.7.mlp.mlp.0.weight
1334
+ dtype: F32
1335
+ shape: [3072, 768]
1336
+ sha256: 74ac79500e66fdac8c6292ddc0554bf9e380f117a065764e88dc22797df85ed3
1337
+ - name: blocks.7.mlp.mlp.2.bias
1338
+ dtype: F32
1339
+ shape: [768]
1340
+ sha256: 260c35d44551bd4fbdc45c4d47b75c7bc7e1ce581a882aec136b573fc78fdcb4
1341
+ - name: blocks.7.mlp.mlp.2.weight
1342
+ dtype: F32
1343
+ shape: [768, 3072]
1344
+ sha256: 6d4ba6b34b0df697470171402aebe7a35eab5cdaea3076a22fa8eca77b10d087
1345
+ - name: blocks.8.attn.to_out.bias
1346
+ dtype: F32
1347
+ shape: [768]
1348
+ sha256: 9f4efdfb33615408edb4a6ddf504a2005c88c6fceecb7aba57727e0e96cc6566
1349
+ - name: blocks.8.attn.to_out.weight
1350
+ dtype: F32
1351
+ shape: [768, 768]
1352
+ sha256: 194cb5e5d3a364052548ea5d82fd6c0fe1393d673b719405fcfab0fc4144ab0e
1353
+ - name: blocks.8.attn.to_qkv.bias
1354
+ dtype: F32
1355
+ shape: [2304]
1356
+ sha256: 342d1946c017bd1f51674943aff35e815080b1c3f22201a5aca2b331ad098a6e
1357
+ - name: blocks.8.attn.to_qkv.weight
1358
+ dtype: F32
1359
+ shape: [2304, 768]
1360
+ sha256: a6c8cd8deb892017f9f9742d392331825580ccdb7dfe0ef3c3bc01697d20caf9
1361
+ - name: blocks.8.mlp.mlp.0.bias
1362
+ dtype: F32
1363
+ shape: [3072]
1364
+ sha256: 22b4d21c72b49da8b3ae2cc5d988cd2f47c82e20b93371b24fce4f3397ec01d7
1365
+ - name: blocks.8.mlp.mlp.0.weight
1366
+ dtype: F32
1367
+ shape: [3072, 768]
1368
+ sha256: df9c9526c04fa9f362a0e2667add15064c99f84a4d0a95a5a934b0c1d3917892
1369
+ - name: blocks.8.mlp.mlp.2.bias
1370
+ dtype: F32
1371
+ shape: [768]
1372
+ sha256: abb392c8709588a69af1803ca987f5f601fce93f2c35316be210b2cdeea0a296
1373
+ - name: blocks.8.mlp.mlp.2.weight
1374
+ dtype: F32
1375
+ shape: [768, 3072]
1376
+ sha256: 8a73fe58624dae8cbe0660ec82b0afbfec579422f4858cc26c49e989a2cbb951
1377
+ - name: blocks.9.attn.to_out.bias
1378
+ dtype: F32
1379
+ shape: [768]
1380
+ sha256: 34a0c7b67055fa65a08753a74169646f6f91208790f0ba6d0b5a5c5514b5ce88
1381
+ - name: blocks.9.attn.to_out.weight
1382
+ dtype: F32
1383
+ shape: [768, 768]
1384
+ sha256: 2b0250f3ba327f80a2febd05ee62ba803bba4d83e3f74e347fcba76dc2c9d622
1385
+ - name: blocks.9.attn.to_qkv.bias
1386
+ dtype: F32
1387
+ shape: [2304]
1388
+ sha256: f8ca458da2cccbf751c1b78f4df7fa6d14d4fa142198f9f15e2259b9ac927a77
1389
+ - name: blocks.9.attn.to_qkv.weight
1390
+ dtype: F32
1391
+ shape: [2304, 768]
1392
+ sha256: 5db75c7b971d684a029a8ad97ea20408fc6665f2451627707668b9fa44be1da2
1393
+ - name: blocks.9.mlp.mlp.0.bias
1394
+ dtype: F32
1395
+ shape: [3072]
1396
+ sha256: 9486a199757cfd5ea8f125077010b07e2903e53a962d30ffc5c8286b8355c610
1397
+ - name: blocks.9.mlp.mlp.0.weight
1398
+ dtype: F32
1399
+ shape: [3072, 768]
1400
+ sha256: 66a4a9a09ed84dc5c17e334cfed1513af6492d2b7aa4d66380a04d55b635592d
1401
+ - name: blocks.9.mlp.mlp.2.bias
1402
+ dtype: F32
1403
+ shape: [768]
1404
+ sha256: 34bdc6b64c7edef3d3d9e937d53109eb32de2fc36b6efb77ee08fbe95970e3f8
1405
+ - name: blocks.9.mlp.mlp.2.weight
1406
+ dtype: F32
1407
+ shape: [768, 3072]
1408
+ sha256: cab92b6cfc51093f301e4f0b87373532e0c437e321d86108a42c11b4ff8d42c2
1409
+ - name: input_layer.bias
1410
+ dtype: F32
1411
+ shape: [768]
1412
+ sha256: de3e799443564d4fa4cf03f958c7a00b2a7c2032c2a00b1206cd9f382dd9aef0
1413
+ - name: input_layer.weight
1414
+ dtype: F32
1415
+ shape: [768, 8]
1416
+ sha256: 1df098322159dad6da1a65420fc56974a4cb7136053f548eb94d8097b08710e2
1417
+ - name: offset_perturbation
1418
+ dtype: F32
1419
+ shape: [4, 3]
1420
+ sha256: f883da8112122d038b97290c6cc46e4906685eab645b05374bb975eaf43ade09
1421
+ - name: out_layer.bias
1422
+ dtype: F32
1423
+ shape: [56]
1424
+ sha256: 4ac03fdac5ac517231b483069e55371dd03609130707838d8f2727f3736c21b2
1425
+ - name: out_layer.weight
1426
+ dtype: F32
1427
+ shape: [56, 768]
1428
+ sha256: 002dac25d67d756586d5448f08c4f792552fdb79bcdd66b6ed2a675fdb2dcf18
checkpoints/conversion_metadata/slat_decoder_mesh.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ source_sha256: 85907b37b67d8ce5b099a96629bdcfbd873eb407dee6b3aa9a75deb15038db33
2
+ converter: mlx-spatial-restricted-torch-zip
3
+ tensor_count: 120
checkpoints/conversion_metadata/slat_generator.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ source_sha256: 91529bde8e7daa12d09618a66c319e3a5a6398db6b23b958cedcb1c3f28faabb
2
+ converter: mlx-spatial-restricted-torch-zip
3
+ tensor_count: 1225
checkpoints/conversion_metadata/ss_decoder.yaml ADDED
@@ -0,0 +1,1135 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ format_version: 1
2
+ source_file: weights/sam-3d-objects/checkpoints/ss_decoder.ckpt
3
+ source_sha256: 6dac1cd7b7fda5a38e0614fadae441f1794f80e39ea2981f1ac8aff0a7e99340
4
+ safetensors_file: ss_decoder.safetensors
5
+ safetensors_files: {}
6
+ created_at_unix: 1777985860
7
+ tensor_count: 74
8
+ total_tensor_bytes: 294684804
9
+ metadata:
10
+ input_layer.weight:
11
+ $type: tensor_ref
12
+ dtype: F32
13
+ storage_key: '0'
14
+ offset_elems: 0
15
+ shape:
16
+ - 512
17
+ - 8
18
+ - 3
19
+ - 3
20
+ - 3
21
+ stride:
22
+ - 216
23
+ - 27
24
+ - 9
25
+ - 3
26
+ - 1
27
+ input_layer.bias:
28
+ $type: tensor_ref
29
+ dtype: F32
30
+ storage_key: '1'
31
+ offset_elems: 0
32
+ shape:
33
+ - 512
34
+ stride:
35
+ - 1
36
+ middle_block.0.norm1.weight:
37
+ $type: tensor_ref
38
+ dtype: F32
39
+ storage_key: '2'
40
+ offset_elems: 0
41
+ shape:
42
+ - 512
43
+ stride:
44
+ - 1
45
+ middle_block.0.norm1.bias:
46
+ $type: tensor_ref
47
+ dtype: F32
48
+ storage_key: '3'
49
+ offset_elems: 0
50
+ shape:
51
+ - 512
52
+ stride:
53
+ - 1
54
+ middle_block.0.norm2.weight:
55
+ $type: tensor_ref
56
+ dtype: F32
57
+ storage_key: '4'
58
+ offset_elems: 0
59
+ shape:
60
+ - 512
61
+ stride:
62
+ - 1
63
+ middle_block.0.norm2.bias:
64
+ $type: tensor_ref
65
+ dtype: F32
66
+ storage_key: '5'
67
+ offset_elems: 0
68
+ shape:
69
+ - 512
70
+ stride:
71
+ - 1
72
+ middle_block.0.conv1.weight:
73
+ $type: tensor_ref
74
+ dtype: F16
75
+ storage_key: '6'
76
+ offset_elems: 0
77
+ shape:
78
+ - 512
79
+ - 512
80
+ - 3
81
+ - 3
82
+ - 3
83
+ stride:
84
+ - 13824
85
+ - 27
86
+ - 9
87
+ - 3
88
+ - 1
89
+ middle_block.0.conv1.bias:
90
+ $type: tensor_ref
91
+ dtype: F16
92
+ storage_key: '7'
93
+ offset_elems: 0
94
+ shape:
95
+ - 512
96
+ stride:
97
+ - 1
98
+ middle_block.0.conv2.weight:
99
+ $type: tensor_ref
100
+ dtype: F16
101
+ storage_key: '8'
102
+ offset_elems: 0
103
+ shape:
104
+ - 512
105
+ - 512
106
+ - 3
107
+ - 3
108
+ - 3
109
+ stride:
110
+ - 13824
111
+ - 27
112
+ - 9
113
+ - 3
114
+ - 1
115
+ middle_block.0.conv2.bias:
116
+ $type: tensor_ref
117
+ dtype: F16
118
+ storage_key: '9'
119
+ offset_elems: 0
120
+ shape:
121
+ - 512
122
+ stride:
123
+ - 1
124
+ middle_block.1.norm1.weight:
125
+ $type: tensor_ref
126
+ dtype: F32
127
+ storage_key: '10'
128
+ offset_elems: 0
129
+ shape:
130
+ - 512
131
+ stride:
132
+ - 1
133
+ middle_block.1.norm1.bias:
134
+ $type: tensor_ref
135
+ dtype: F32
136
+ storage_key: '11'
137
+ offset_elems: 0
138
+ shape:
139
+ - 512
140
+ stride:
141
+ - 1
142
+ middle_block.1.norm2.weight:
143
+ $type: tensor_ref
144
+ dtype: F32
145
+ storage_key: '12'
146
+ offset_elems: 0
147
+ shape:
148
+ - 512
149
+ stride:
150
+ - 1
151
+ middle_block.1.norm2.bias:
152
+ $type: tensor_ref
153
+ dtype: F32
154
+ storage_key: '13'
155
+ offset_elems: 0
156
+ shape:
157
+ - 512
158
+ stride:
159
+ - 1
160
+ middle_block.1.conv1.weight:
161
+ $type: tensor_ref
162
+ dtype: F16
163
+ storage_key: '14'
164
+ offset_elems: 0
165
+ shape:
166
+ - 512
167
+ - 512
168
+ - 3
169
+ - 3
170
+ - 3
171
+ stride:
172
+ - 13824
173
+ - 27
174
+ - 9
175
+ - 3
176
+ - 1
177
+ middle_block.1.conv1.bias:
178
+ $type: tensor_ref
179
+ dtype: F16
180
+ storage_key: '15'
181
+ offset_elems: 0
182
+ shape:
183
+ - 512
184
+ stride:
185
+ - 1
186
+ middle_block.1.conv2.weight:
187
+ $type: tensor_ref
188
+ dtype: F16
189
+ storage_key: '16'
190
+ offset_elems: 0
191
+ shape:
192
+ - 512
193
+ - 512
194
+ - 3
195
+ - 3
196
+ - 3
197
+ stride:
198
+ - 13824
199
+ - 27
200
+ - 9
201
+ - 3
202
+ - 1
203
+ middle_block.1.conv2.bias:
204
+ $type: tensor_ref
205
+ dtype: F16
206
+ storage_key: '17'
207
+ offset_elems: 0
208
+ shape:
209
+ - 512
210
+ stride:
211
+ - 1
212
+ blocks.0.norm1.weight:
213
+ $type: tensor_ref
214
+ dtype: F32
215
+ storage_key: '18'
216
+ offset_elems: 0
217
+ shape:
218
+ - 512
219
+ stride:
220
+ - 1
221
+ blocks.0.norm1.bias:
222
+ $type: tensor_ref
223
+ dtype: F32
224
+ storage_key: '19'
225
+ offset_elems: 0
226
+ shape:
227
+ - 512
228
+ stride:
229
+ - 1
230
+ blocks.0.norm2.weight:
231
+ $type: tensor_ref
232
+ dtype: F32
233
+ storage_key: '20'
234
+ offset_elems: 0
235
+ shape:
236
+ - 512
237
+ stride:
238
+ - 1
239
+ blocks.0.norm2.bias:
240
+ $type: tensor_ref
241
+ dtype: F32
242
+ storage_key: '21'
243
+ offset_elems: 0
244
+ shape:
245
+ - 512
246
+ stride:
247
+ - 1
248
+ blocks.0.conv1.weight:
249
+ $type: tensor_ref
250
+ dtype: F16
251
+ storage_key: '22'
252
+ offset_elems: 0
253
+ shape:
254
+ - 512
255
+ - 512
256
+ - 3
257
+ - 3
258
+ - 3
259
+ stride:
260
+ - 13824
261
+ - 27
262
+ - 9
263
+ - 3
264
+ - 1
265
+ blocks.0.conv1.bias:
266
+ $type: tensor_ref
267
+ dtype: F16
268
+ storage_key: '23'
269
+ offset_elems: 0
270
+ shape:
271
+ - 512
272
+ stride:
273
+ - 1
274
+ blocks.0.conv2.weight:
275
+ $type: tensor_ref
276
+ dtype: F16
277
+ storage_key: '24'
278
+ offset_elems: 0
279
+ shape:
280
+ - 512
281
+ - 512
282
+ - 3
283
+ - 3
284
+ - 3
285
+ stride:
286
+ - 13824
287
+ - 27
288
+ - 9
289
+ - 3
290
+ - 1
291
+ blocks.0.conv2.bias:
292
+ $type: tensor_ref
293
+ dtype: F16
294
+ storage_key: '25'
295
+ offset_elems: 0
296
+ shape:
297
+ - 512
298
+ stride:
299
+ - 1
300
+ blocks.1.norm1.weight:
301
+ $type: tensor_ref
302
+ dtype: F32
303
+ storage_key: '26'
304
+ offset_elems: 0
305
+ shape:
306
+ - 512
307
+ stride:
308
+ - 1
309
+ blocks.1.norm1.bias:
310
+ $type: tensor_ref
311
+ dtype: F32
312
+ storage_key: '27'
313
+ offset_elems: 0
314
+ shape:
315
+ - 512
316
+ stride:
317
+ - 1
318
+ blocks.1.norm2.weight:
319
+ $type: tensor_ref
320
+ dtype: F32
321
+ storage_key: '28'
322
+ offset_elems: 0
323
+ shape:
324
+ - 512
325
+ stride:
326
+ - 1
327
+ blocks.1.norm2.bias:
328
+ $type: tensor_ref
329
+ dtype: F32
330
+ storage_key: '29'
331
+ offset_elems: 0
332
+ shape:
333
+ - 512
334
+ stride:
335
+ - 1
336
+ blocks.1.conv1.weight:
337
+ $type: tensor_ref
338
+ dtype: F16
339
+ storage_key: '30'
340
+ offset_elems: 0
341
+ shape:
342
+ - 512
343
+ - 512
344
+ - 3
345
+ - 3
346
+ - 3
347
+ stride:
348
+ - 13824
349
+ - 27
350
+ - 9
351
+ - 3
352
+ - 1
353
+ blocks.1.conv1.bias:
354
+ $type: tensor_ref
355
+ dtype: F16
356
+ storage_key: '31'
357
+ offset_elems: 0
358
+ shape:
359
+ - 512
360
+ stride:
361
+ - 1
362
+ blocks.1.conv2.weight:
363
+ $type: tensor_ref
364
+ dtype: F16
365
+ storage_key: '32'
366
+ offset_elems: 0
367
+ shape:
368
+ - 512
369
+ - 512
370
+ - 3
371
+ - 3
372
+ - 3
373
+ stride:
374
+ - 13824
375
+ - 27
376
+ - 9
377
+ - 3
378
+ - 1
379
+ blocks.1.conv2.bias:
380
+ $type: tensor_ref
381
+ dtype: F16
382
+ storage_key: '33'
383
+ offset_elems: 0
384
+ shape:
385
+ - 512
386
+ stride:
387
+ - 1
388
+ blocks.2.conv.weight:
389
+ $type: tensor_ref
390
+ dtype: F16
391
+ storage_key: '34'
392
+ offset_elems: 0
393
+ shape:
394
+ - 1024
395
+ - 512
396
+ - 3
397
+ - 3
398
+ - 3
399
+ stride:
400
+ - 13824
401
+ - 27
402
+ - 9
403
+ - 3
404
+ - 1
405
+ blocks.2.conv.bias:
406
+ $type: tensor_ref
407
+ dtype: F16
408
+ storage_key: '35'
409
+ offset_elems: 0
410
+ shape:
411
+ - 1024
412
+ stride:
413
+ - 1
414
+ blocks.3.norm1.weight:
415
+ $type: tensor_ref
416
+ dtype: F32
417
+ storage_key: '36'
418
+ offset_elems: 0
419
+ shape:
420
+ - 128
421
+ stride:
422
+ - 1
423
+ blocks.3.norm1.bias:
424
+ $type: tensor_ref
425
+ dtype: F32
426
+ storage_key: '37'
427
+ offset_elems: 0
428
+ shape:
429
+ - 128
430
+ stride:
431
+ - 1
432
+ blocks.3.norm2.weight:
433
+ $type: tensor_ref
434
+ dtype: F32
435
+ storage_key: '38'
436
+ offset_elems: 0
437
+ shape:
438
+ - 128
439
+ stride:
440
+ - 1
441
+ blocks.3.norm2.bias:
442
+ $type: tensor_ref
443
+ dtype: F32
444
+ storage_key: '39'
445
+ offset_elems: 0
446
+ shape:
447
+ - 128
448
+ stride:
449
+ - 1
450
+ blocks.3.conv1.weight:
451
+ $type: tensor_ref
452
+ dtype: F16
453
+ storage_key: '40'
454
+ offset_elems: 0
455
+ shape:
456
+ - 128
457
+ - 128
458
+ - 3
459
+ - 3
460
+ - 3
461
+ stride:
462
+ - 3456
463
+ - 27
464
+ - 9
465
+ - 3
466
+ - 1
467
+ blocks.3.conv1.bias:
468
+ $type: tensor_ref
469
+ dtype: F16
470
+ storage_key: '41'
471
+ offset_elems: 0
472
+ shape:
473
+ - 128
474
+ stride:
475
+ - 1
476
+ blocks.3.conv2.weight:
477
+ $type: tensor_ref
478
+ dtype: F16
479
+ storage_key: '42'
480
+ offset_elems: 0
481
+ shape:
482
+ - 128
483
+ - 128
484
+ - 3
485
+ - 3
486
+ - 3
487
+ stride:
488
+ - 3456
489
+ - 27
490
+ - 9
491
+ - 3
492
+ - 1
493
+ blocks.3.conv2.bias:
494
+ $type: tensor_ref
495
+ dtype: F16
496
+ storage_key: '43'
497
+ offset_elems: 0
498
+ shape:
499
+ - 128
500
+ stride:
501
+ - 1
502
+ blocks.4.norm1.weight:
503
+ $type: tensor_ref
504
+ dtype: F32
505
+ storage_key: '44'
506
+ offset_elems: 0
507
+ shape:
508
+ - 128
509
+ stride:
510
+ - 1
511
+ blocks.4.norm1.bias:
512
+ $type: tensor_ref
513
+ dtype: F32
514
+ storage_key: '45'
515
+ offset_elems: 0
516
+ shape:
517
+ - 128
518
+ stride:
519
+ - 1
520
+ blocks.4.norm2.weight:
521
+ $type: tensor_ref
522
+ dtype: F32
523
+ storage_key: '46'
524
+ offset_elems: 0
525
+ shape:
526
+ - 128
527
+ stride:
528
+ - 1
529
+ blocks.4.norm2.bias:
530
+ $type: tensor_ref
531
+ dtype: F32
532
+ storage_key: '47'
533
+ offset_elems: 0
534
+ shape:
535
+ - 128
536
+ stride:
537
+ - 1
538
+ blocks.4.conv1.weight:
539
+ $type: tensor_ref
540
+ dtype: F16
541
+ storage_key: '48'
542
+ offset_elems: 0
543
+ shape:
544
+ - 128
545
+ - 128
546
+ - 3
547
+ - 3
548
+ - 3
549
+ stride:
550
+ - 3456
551
+ - 27
552
+ - 9
553
+ - 3
554
+ - 1
555
+ blocks.4.conv1.bias:
556
+ $type: tensor_ref
557
+ dtype: F16
558
+ storage_key: '49'
559
+ offset_elems: 0
560
+ shape:
561
+ - 128
562
+ stride:
563
+ - 1
564
+ blocks.4.conv2.weight:
565
+ $type: tensor_ref
566
+ dtype: F16
567
+ storage_key: '50'
568
+ offset_elems: 0
569
+ shape:
570
+ - 128
571
+ - 128
572
+ - 3
573
+ - 3
574
+ - 3
575
+ stride:
576
+ - 3456
577
+ - 27
578
+ - 9
579
+ - 3
580
+ - 1
581
+ blocks.4.conv2.bias:
582
+ $type: tensor_ref
583
+ dtype: F16
584
+ storage_key: '51'
585
+ offset_elems: 0
586
+ shape:
587
+ - 128
588
+ stride:
589
+ - 1
590
+ blocks.5.conv.weight:
591
+ $type: tensor_ref
592
+ dtype: F16
593
+ storage_key: '52'
594
+ offset_elems: 0
595
+ shape:
596
+ - 256
597
+ - 128
598
+ - 3
599
+ - 3
600
+ - 3
601
+ stride:
602
+ - 3456
603
+ - 27
604
+ - 9
605
+ - 3
606
+ - 1
607
+ blocks.5.conv.bias:
608
+ $type: tensor_ref
609
+ dtype: F16
610
+ storage_key: '53'
611
+ offset_elems: 0
612
+ shape:
613
+ - 256
614
+ stride:
615
+ - 1
616
+ blocks.6.norm1.weight:
617
+ $type: tensor_ref
618
+ dtype: F32
619
+ storage_key: '54'
620
+ offset_elems: 0
621
+ shape:
622
+ - 32
623
+ stride:
624
+ - 1
625
+ blocks.6.norm1.bias:
626
+ $type: tensor_ref
627
+ dtype: F32
628
+ storage_key: '55'
629
+ offset_elems: 0
630
+ shape:
631
+ - 32
632
+ stride:
633
+ - 1
634
+ blocks.6.norm2.weight:
635
+ $type: tensor_ref
636
+ dtype: F32
637
+ storage_key: '56'
638
+ offset_elems: 0
639
+ shape:
640
+ - 32
641
+ stride:
642
+ - 1
643
+ blocks.6.norm2.bias:
644
+ $type: tensor_ref
645
+ dtype: F32
646
+ storage_key: '57'
647
+ offset_elems: 0
648
+ shape:
649
+ - 32
650
+ stride:
651
+ - 1
652
+ blocks.6.conv1.weight:
653
+ $type: tensor_ref
654
+ dtype: F16
655
+ storage_key: '58'
656
+ offset_elems: 0
657
+ shape:
658
+ - 32
659
+ - 32
660
+ - 3
661
+ - 3
662
+ - 3
663
+ stride:
664
+ - 864
665
+ - 27
666
+ - 9
667
+ - 3
668
+ - 1
669
+ blocks.6.conv1.bias:
670
+ $type: tensor_ref
671
+ dtype: F16
672
+ storage_key: '59'
673
+ offset_elems: 0
674
+ shape:
675
+ - 32
676
+ stride:
677
+ - 1
678
+ blocks.6.conv2.weight:
679
+ $type: tensor_ref
680
+ dtype: F16
681
+ storage_key: '60'
682
+ offset_elems: 0
683
+ shape:
684
+ - 32
685
+ - 32
686
+ - 3
687
+ - 3
688
+ - 3
689
+ stride:
690
+ - 864
691
+ - 27
692
+ - 9
693
+ - 3
694
+ - 1
695
+ blocks.6.conv2.bias:
696
+ $type: tensor_ref
697
+ dtype: F16
698
+ storage_key: '61'
699
+ offset_elems: 0
700
+ shape:
701
+ - 32
702
+ stride:
703
+ - 1
704
+ blocks.7.norm1.weight:
705
+ $type: tensor_ref
706
+ dtype: F32
707
+ storage_key: '62'
708
+ offset_elems: 0
709
+ shape:
710
+ - 32
711
+ stride:
712
+ - 1
713
+ blocks.7.norm1.bias:
714
+ $type: tensor_ref
715
+ dtype: F32
716
+ storage_key: '63'
717
+ offset_elems: 0
718
+ shape:
719
+ - 32
720
+ stride:
721
+ - 1
722
+ blocks.7.norm2.weight:
723
+ $type: tensor_ref
724
+ dtype: F32
725
+ storage_key: '64'
726
+ offset_elems: 0
727
+ shape:
728
+ - 32
729
+ stride:
730
+ - 1
731
+ blocks.7.norm2.bias:
732
+ $type: tensor_ref
733
+ dtype: F32
734
+ storage_key: '65'
735
+ offset_elems: 0
736
+ shape:
737
+ - 32
738
+ stride:
739
+ - 1
740
+ blocks.7.conv1.weight:
741
+ $type: tensor_ref
742
+ dtype: F16
743
+ storage_key: '66'
744
+ offset_elems: 0
745
+ shape:
746
+ - 32
747
+ - 32
748
+ - 3
749
+ - 3
750
+ - 3
751
+ stride:
752
+ - 864
753
+ - 27
754
+ - 9
755
+ - 3
756
+ - 1
757
+ blocks.7.conv1.bias:
758
+ $type: tensor_ref
759
+ dtype: F16
760
+ storage_key: '67'
761
+ offset_elems: 0
762
+ shape:
763
+ - 32
764
+ stride:
765
+ - 1
766
+ blocks.7.conv2.weight:
767
+ $type: tensor_ref
768
+ dtype: F16
769
+ storage_key: '68'
770
+ offset_elems: 0
771
+ shape:
772
+ - 32
773
+ - 32
774
+ - 3
775
+ - 3
776
+ - 3
777
+ stride:
778
+ - 864
779
+ - 27
780
+ - 9
781
+ - 3
782
+ - 1
783
+ blocks.7.conv2.bias:
784
+ $type: tensor_ref
785
+ dtype: F16
786
+ storage_key: '69'
787
+ offset_elems: 0
788
+ shape:
789
+ - 32
790
+ stride:
791
+ - 1
792
+ out_layer.0.weight:
793
+ $type: tensor_ref
794
+ dtype: F32
795
+ storage_key: '70'
796
+ offset_elems: 0
797
+ shape:
798
+ - 32
799
+ stride:
800
+ - 1
801
+ out_layer.0.bias:
802
+ $type: tensor_ref
803
+ dtype: F32
804
+ storage_key: '71'
805
+ offset_elems: 0
806
+ shape:
807
+ - 32
808
+ stride:
809
+ - 1
810
+ out_layer.2.weight:
811
+ $type: tensor_ref
812
+ dtype: F32
813
+ storage_key: '72'
814
+ offset_elems: 0
815
+ shape:
816
+ - 1
817
+ - 32
818
+ - 3
819
+ - 3
820
+ - 3
821
+ stride:
822
+ - 864
823
+ - 27
824
+ - 9
825
+ - 3
826
+ - 1
827
+ out_layer.2.bias:
828
+ $type: tensor_ref
829
+ dtype: F32
830
+ storage_key: '73'
831
+ offset_elems: 0
832
+ shape:
833
+ - 1
834
+ stride:
835
+ - 1
836
+ security:
837
+ objects: []
838
+ calls: []
839
+ tensors:
840
+ - name: blocks.0.conv1.bias
841
+ dtype: F32
842
+ shape: [512]
843
+ sha256: 8318dd61f3a2f63c3772c6479f225124860c22416087a74f67f60bfe968953a7
844
+ - name: blocks.0.conv1.weight
845
+ dtype: F32
846
+ shape: [512, 512, 3, 3, 3]
847
+ sha256: 01ff17526cc23c623ce7591a4984083060289011cf6cae5da81d4eb1caadaa48
848
+ - name: blocks.0.conv2.bias
849
+ dtype: F32
850
+ shape: [512]
851
+ sha256: fcd4a3e37b1cf4a743ce19b421591280647e7f70ad8750e1e4fbc55486baf8fe
852
+ - name: blocks.0.conv2.weight
853
+ dtype: F32
854
+ shape: [512, 512, 3, 3, 3]
855
+ sha256: b5d8773f6eab8620eae01c38e24690306989393cabd3272b57e698ed9f3af489
856
+ - name: blocks.0.norm1.bias
857
+ dtype: F32
858
+ shape: [512]
859
+ sha256: 5929b89d4637b31ee4437af03714939bbff60d844d5d6d79bac6a13ae86b489e
860
+ - name: blocks.0.norm1.weight
861
+ dtype: F32
862
+ shape: [512]
863
+ sha256: 2e1637708bee73d04d573756acb299139ad0a6d341e700184955f64d89779116
864
+ - name: blocks.0.norm2.bias
865
+ dtype: F32
866
+ shape: [512]
867
+ sha256: 53be0083949fd4b2d312676041191fcf5d27ecc22865ef503dd5a6fe9aaa376b
868
+ - name: blocks.0.norm2.weight
869
+ dtype: F32
870
+ shape: [512]
871
+ sha256: 1d063ef48dccdbf623e6a3fdeefb42eb46a0fd262713915d9a38e273825c30ec
872
+ - name: blocks.1.conv1.bias
873
+ dtype: F32
874
+ shape: [512]
875
+ sha256: 501ca55f325c10927c6887e34f2b05de85a39601c18732b08bf7aa9d7e021724
876
+ - name: blocks.1.conv1.weight
877
+ dtype: F32
878
+ shape: [512, 512, 3, 3, 3]
879
+ sha256: 43bd691f53b02c8de9c59117bcf4609f271e69f288d311c5cf2e0f55fec6a3fd
880
+ - name: blocks.1.conv2.bias
881
+ dtype: F32
882
+ shape: [512]
883
+ sha256: 0c43c6bc561212b1b1b51e26c05967767d17f91444f5f6a17733bc591bc26703
884
+ - name: blocks.1.conv2.weight
885
+ dtype: F32
886
+ shape: [512, 512, 3, 3, 3]
887
+ sha256: 94187be51a47542857840ec0b6cd65d25dcb50e6a97c02bd0b5c33fda2735b72
888
+ - name: blocks.1.norm1.bias
889
+ dtype: F32
890
+ shape: [512]
891
+ sha256: a81c8a5a65ca89193809f084319f4f336efe14e2e71d4c4d7f54e64f5370c46e
892
+ - name: blocks.1.norm1.weight
893
+ dtype: F32
894
+ shape: [512]
895
+ sha256: b817ab88b36eef7b4b5daa3300a0567debbec0dcd227d8633126c67082b35254
896
+ - name: blocks.1.norm2.bias
897
+ dtype: F32
898
+ shape: [512]
899
+ sha256: abd3195ab109b3bead4b06d9a3273a3d374d8361e62e6ce2b6fc3080584f68ba
900
+ - name: blocks.1.norm2.weight
901
+ dtype: F32
902
+ shape: [512]
903
+ sha256: a3699af15300e0f09fe586defa66711d3004901d191e8b703987b5728e98ffa0
904
+ - name: blocks.2.conv.bias
905
+ dtype: F32
906
+ shape: [1024]
907
+ sha256: c241b549930db6829cc5a1457b3d861938146fa4aff63fb23322f62964f9f8d3
908
+ - name: blocks.2.conv.weight
909
+ dtype: F32
910
+ shape: [1024, 512, 3, 3, 3]
911
+ sha256: 15f4e22aac51c9fd089881d4b6b0f0633bef095b8e0199691f0bcd10d9950bfe
912
+ - name: blocks.3.conv1.bias
913
+ dtype: F32
914
+ shape: [128]
915
+ sha256: 3e4d87e5e680f0f9f52ea32cd45e001c81449bb48032f87a578d72015638c965
916
+ - name: blocks.3.conv1.weight
917
+ dtype: F32
918
+ shape: [128, 128, 3, 3, 3]
919
+ sha256: e5c1fd3a1891bffcce0c72062f75b5ea0c7b2deb954c695e9bd1b80f9b02ea15
920
+ - name: blocks.3.conv2.bias
921
+ dtype: F32
922
+ shape: [128]
923
+ sha256: 84eeb5105a4c53f05a668fc278e07739e7a1df04c400003b329b5d606a1810cf
924
+ - name: blocks.3.conv2.weight
925
+ dtype: F32
926
+ shape: [128, 128, 3, 3, 3]
927
+ sha256: faa6ed1fe82675e292b047ccf142006df130821963aeb0f5cb0baa27b156a24f
928
+ - name: blocks.3.norm1.bias
929
+ dtype: F32
930
+ shape: [128]
931
+ sha256: fdb6ce03e9b1abe1d2ddf121baa27932b80b51f388067b6a6e28f3c35559f9f0
932
+ - name: blocks.3.norm1.weight
933
+ dtype: F32
934
+ shape: [128]
935
+ sha256: 73a13ad3c9b4c456273bd4f6270d4ec7c4a760a6b5f0db49f54120ea5b53aec6
936
+ - name: blocks.3.norm2.bias
937
+ dtype: F32
938
+ shape: [128]
939
+ sha256: 9d35644516579187f06c2694b95301ff3311685e57dd4c7fa0c0f30afc59f1e8
940
+ - name: blocks.3.norm2.weight
941
+ dtype: F32
942
+ shape: [128]
943
+ sha256: 105843929d5f9312a2d2e541db8dedf75a5ae5db1ce722dcc1ca31206cf11463
944
+ - name: blocks.4.conv1.bias
945
+ dtype: F32
946
+ shape: [128]
947
+ sha256: 4dbfb743b4b0c482c2284cc403b8ae8fd503a4008d59c00432adb244dae451f0
948
+ - name: blocks.4.conv1.weight
949
+ dtype: F32
950
+ shape: [128, 128, 3, 3, 3]
951
+ sha256: ec3b57e14fa39b11ddbf666239a9ba852abb509dd0b3ffeaff007801f9ee1a2d
952
+ - name: blocks.4.conv2.bias
953
+ dtype: F32
954
+ shape: [128]
955
+ sha256: 1407083cb0f4b1ceda4d865ba6603f0966ed2ca9aa861e16d66430f241cd7152
956
+ - name: blocks.4.conv2.weight
957
+ dtype: F32
958
+ shape: [128, 128, 3, 3, 3]
959
+ sha256: d2603cd233065cfb128a72f4c5caab180c495866801d3ac8cf3a31996559f3dd
960
+ - name: blocks.4.norm1.bias
961
+ dtype: F32
962
+ shape: [128]
963
+ sha256: 3064c944d8def8d7a727dbd621b60b727dcca7f1e3d86a56a5dfc02f10d374cc
964
+ - name: blocks.4.norm1.weight
965
+ dtype: F32
966
+ shape: [128]
967
+ sha256: aff245257ce6f9c1cf65c9fbbbb6d6a5fe63b9e42647ca6e778675f3a68d7338
968
+ - name: blocks.4.norm2.bias
969
+ dtype: F32
970
+ shape: [128]
971
+ sha256: ace1030bbb55a1750b1ca3e8572b7378b95cc211c0c9ec80dcb63d51f0bae107
972
+ - name: blocks.4.norm2.weight
973
+ dtype: F32
974
+ shape: [128]
975
+ sha256: 2afc00508d28cbd360ffb4400fd230f58ac57778b9eecc170589f79fcbb8ec2d
976
+ - name: blocks.5.conv.bias
977
+ dtype: F32
978
+ shape: [256]
979
+ sha256: 2ab3516c2fec4aac1c387d923bf3543fde3be0edae6d279819efd43897f2dbad
980
+ - name: blocks.5.conv.weight
981
+ dtype: F32
982
+ shape: [256, 128, 3, 3, 3]
983
+ sha256: 2556506e3f1d5ee38e6a4f61aa19e7fc87f7542f8bbcd15c6e27103a18e88dc1
984
+ - name: blocks.6.conv1.bias
985
+ dtype: F32
986
+ shape: [32]
987
+ sha256: 309c5e0880cea7b7d9835682caea01ada62b7f4f88379132b16c2b531b9241bb
988
+ - name: blocks.6.conv1.weight
989
+ dtype: F32
990
+ shape: [32, 32, 3, 3, 3]
991
+ sha256: 49d1b048eca46babfece2c5ae4aab3a5d03eb95e93cc1bfcef5e001624812063
992
+ - name: blocks.6.conv2.bias
993
+ dtype: F32
994
+ shape: [32]
995
+ sha256: 3ce6fd5bb17b79a9d2a6bf25ab90d409f6d52859348e2acf9a92074023261ef7
996
+ - name: blocks.6.conv2.weight
997
+ dtype: F32
998
+ shape: [32, 32, 3, 3, 3]
999
+ sha256: c098477dc69229cd1aa5aac2e458965eb046e534797e453c5c06dff7c622fbac
1000
+ - name: blocks.6.norm1.bias
1001
+ dtype: F32
1002
+ shape: [32]
1003
+ sha256: 1c580c9b374539a967333d51e6aa6ceb958a240744a30688011da3355454a151
1004
+ - name: blocks.6.norm1.weight
1005
+ dtype: F32
1006
+ shape: [32]
1007
+ sha256: 511e41d2b06596e60432c556e3d592f80886399b4de8cbc6bbd2b72ced6d4cca
1008
+ - name: blocks.6.norm2.bias
1009
+ dtype: F32
1010
+ shape: [32]
1011
+ sha256: e823da2e0bee9dd00c900d3c2946d95d3ac673dde6d623a0a8d2106161da1900
1012
+ - name: blocks.6.norm2.weight
1013
+ dtype: F32
1014
+ shape: [32]
1015
+ sha256: 913cce9c09e37b8d0109e863071cd4787bcf5c1d82d10e79f58bc128e27fe39a
1016
+ - name: blocks.7.conv1.bias
1017
+ dtype: F32
1018
+ shape: [32]
1019
+ sha256: 0fbdbf36d8d0e5b0fc41efefd87aa5035d88b5241cc80c21a57bb0e3c9f2c8c2
1020
+ - name: blocks.7.conv1.weight
1021
+ dtype: F32
1022
+ shape: [32, 32, 3, 3, 3]
1023
+ sha256: f00ff5a16ee0cc86a19aa58eac96005388c57e911a258082d0e813f926d73666
1024
+ - name: blocks.7.conv2.bias
1025
+ dtype: F32
1026
+ shape: [32]
1027
+ sha256: 4ecfeddaf3e16219a6f05642c960615d38c4c37e90c2a195e8b38d909a873f41
1028
+ - name: blocks.7.conv2.weight
1029
+ dtype: F32
1030
+ shape: [32, 32, 3, 3, 3]
1031
+ sha256: 13550fea8ab17e6dc86ba457ab787427ba34e4117d01bbaa47de49b3635a77c3
1032
+ - name: blocks.7.norm1.bias
1033
+ dtype: F32
1034
+ shape: [32]
1035
+ sha256: 4bdf8d3e828dcc693604714298ece99b89e1f338f80f5d403cf151372f9f25db
1036
+ - name: blocks.7.norm1.weight
1037
+ dtype: F32
1038
+ shape: [32]
1039
+ sha256: 75ee209e1c2835f6c3bfe5a9c22863a879e4ea36577b0e1c777dd3fcd3473506
1040
+ - name: blocks.7.norm2.bias
1041
+ dtype: F32
1042
+ shape: [32]
1043
+ sha256: a6855c6ca25a39de616d27e2da72a9e8d66e1ca8e65049ca30a9cdbc82a25d50
1044
+ - name: blocks.7.norm2.weight
1045
+ dtype: F32
1046
+ shape: [32]
1047
+ sha256: 7275a74bd381e56e9c4e172123fa774b758a0097d2a01ea87fceba20f0aab542
1048
+ - name: input_layer.bias
1049
+ dtype: F32
1050
+ shape: [512]
1051
+ sha256: 02e1ba98d624440ce8b6853744170bf687a2d44545fdb022568f9dfa79791441
1052
+ - name: input_layer.weight
1053
+ dtype: F32
1054
+ shape: [512, 8, 3, 3, 3]
1055
+ sha256: 5d77c0a43d668d5ef7cf87dcd9707f90f1f2d53aaf85d9e13ab9001ffd7e4c60
1056
+ - name: middle_block.0.conv1.bias
1057
+ dtype: F32
1058
+ shape: [512]
1059
+ sha256: 828b56016292b7f5f837d0a069c0ac9425ed6f063e4c59cea254506ea2ae5aca
1060
+ - name: middle_block.0.conv1.weight
1061
+ dtype: F32
1062
+ shape: [512, 512, 3, 3, 3]
1063
+ sha256: a83c00f1634536165b8282fa9d180a7f2b02b3948f82a5c9b58afad059ff9063
1064
+ - name: middle_block.0.conv2.bias
1065
+ dtype: F32
1066
+ shape: [512]
1067
+ sha256: e3d5616e1ac50b9894a3c62e5f00ea9381ca2eef1d195abc1876e95c5b935058
1068
+ - name: middle_block.0.conv2.weight
1069
+ dtype: F32
1070
+ shape: [512, 512, 3, 3, 3]
1071
+ sha256: 645b83032cdfc89a2266a3b0ab0c182dd00bafb6f3f2eb24b4c6ab934cb3a6f1
1072
+ - name: middle_block.0.norm1.bias
1073
+ dtype: F32
1074
+ shape: [512]
1075
+ sha256: 3f0dc8da1cb01b1af817a17f56b1385131e3747bb3c1f5b0a1558b66f432cc30
1076
+ - name: middle_block.0.norm1.weight
1077
+ dtype: F32
1078
+ shape: [512]
1079
+ sha256: a385e123ca8882d97e1086c075b782b3ce0060608c295e2b9daa9c20008ce842
1080
+ - name: middle_block.0.norm2.bias
1081
+ dtype: F32
1082
+ shape: [512]
1083
+ sha256: 208f9d6ed284d8e13862a33229cdedc39f272fac271035a0fa00ce647aacb005
1084
+ - name: middle_block.0.norm2.weight
1085
+ dtype: F32
1086
+ shape: [512]
1087
+ sha256: e484a45edeea7213c7be6f161cae8754a612ec738aaf5b0e80f5ada506c53064
1088
+ - name: middle_block.1.conv1.bias
1089
+ dtype: F32
1090
+ shape: [512]
1091
+ sha256: 5415b21b405fefc85ebc14744c4b0004808bf7fa2793a1164103fbcb1c8e53eb
1092
+ - name: middle_block.1.conv1.weight
1093
+ dtype: F32
1094
+ shape: [512, 512, 3, 3, 3]
1095
+ sha256: 6571348089f9d14df0276ba187e4e51835c8bc6b14f92e68138999844f70483c
1096
+ - name: middle_block.1.conv2.bias
1097
+ dtype: F32
1098
+ shape: [512]
1099
+ sha256: a42db30c6874337a578c507aead4f60c41078c8959176865c706e91ece9b3664
1100
+ - name: middle_block.1.conv2.weight
1101
+ dtype: F32
1102
+ shape: [512, 512, 3, 3, 3]
1103
+ sha256: 9d01d4c9d49b27f2cf2bd0640c22356c6c1ca46e677c0a25132f1c281afb3382
1104
+ - name: middle_block.1.norm1.bias
1105
+ dtype: F32
1106
+ shape: [512]
1107
+ sha256: 3ce8f2e4b093e52835070536961c4bda43fc17d939fee1fb55400acd0298e8d8
1108
+ - name: middle_block.1.norm1.weight
1109
+ dtype: F32
1110
+ shape: [512]
1111
+ sha256: 89aacb9ec0df903f1a10c781cd5a5a43d488020cd4f5506c4b743b0aae47107a
1112
+ - name: middle_block.1.norm2.bias
1113
+ dtype: F32
1114
+ shape: [512]
1115
+ sha256: 73771be88240465e135c10911863731a87afd17f89f87e548eeddcede0c55119
1116
+ - name: middle_block.1.norm2.weight
1117
+ dtype: F32
1118
+ shape: [512]
1119
+ sha256: 43dcb2fa6f11de5d274ff95f978c12d7e4168dc92f2420d90ee9247c4a60ed81
1120
+ - name: out_layer.0.bias
1121
+ dtype: F32
1122
+ shape: [32]
1123
+ sha256: f146a5721011d745734ea69211f411f2eb7f056df47128baba4b8b5fbf96c00a
1124
+ - name: out_layer.0.weight
1125
+ dtype: F32
1126
+ shape: [32]
1127
+ sha256: 0613d82a3a0712a7c44c20ff997f81f596727d5bc00c75fab00389fa5d7f56c0
1128
+ - name: out_layer.2.bias
1129
+ dtype: F32
1130
+ shape: [1]
1131
+ sha256: 03db042d62b7325a8dc4b7f6c10fde3186462335604bdba331c29f2c3e266dcf
1132
+ - name: out_layer.2.weight
1133
+ dtype: F32
1134
+ shape: [1, 32, 3, 3, 3]
1135
+ sha256: c5b357c174e0b9539605b88b9de84c41ab6c238e10cb6c4cbc41708763a2f5b5
checkpoints/conversion_metadata/ss_generator.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ source_sha256: 225f40479e4cff4f39d6fa14c55be3abad1475bf55b61af3bec1e19ed2f6c146
2
+ converter: mlx-spatial-restricted-torch-zip
3
+ tensor_count: 1741
checkpoints/pipeline.yaml ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: sam3d_objects.pipeline.inference_pipeline_pointmap.InferencePipelinePointMap
2
+ ss_generator_config_path: ss_generator.yaml
3
+ ss_generator_ckpt_path: ss_generator.safetensors
4
+ slat_generator_config_path: slat_generator.yaml
5
+ slat_generator_ckpt_path: slat_generator.safetensors
6
+ ss_decoder_config_path: ss_decoder.yaml
7
+ ss_decoder_ckpt_path: ss_decoder.safetensors
8
+ slat_decoder_gs_config_path: slat_decoder_gs.yaml
9
+ slat_decoder_gs_ckpt_path: slat_decoder_gs.safetensors
10
+ slat_decoder_gs_4_config_path: slat_decoder_gs_4.yaml
11
+ slat_decoder_gs_4_ckpt_path: slat_decoder_gs_4.safetensors
12
+ slat_decoder_mesh_config_path: slat_decoder_mesh.yaml
13
+ slat_decoder_mesh_ckpt_path: slat_decoder_mesh.safetensors
14
+ pad_size: 1.0
15
+ dtype: float16
16
+ version: 3dfy_v9
17
+ slat_cfg_strength: 1
18
+ slat_rescale_t: 1
19
+ downsample_ss_dist: 1
20
+ compile_model: true
21
+ ss_condition_input_mapping: []
22
+ ss_preprocessor:
23
+ _target_: sam3d_objects.data.dataset.tdfy.preprocessor.PreProcessor
24
+ img_mask_joint_transform: []
25
+ img_mask_pointmap_joint_transform:
26
+ - _partial_: true
27
+ _target_: sam3d_objects.data.dataset.tdfy.img_and_mask_transforms.resize_all_to_same_size
28
+ - _partial_: true
29
+ _target_: sam3d_objects.data.dataset.tdfy.img_and_mask_transforms.crop_around_mask_with_padding
30
+ box_size_factor: 1.2
31
+ padding_factor: 0.0
32
+ img_transform:
33
+ _target_: torchvision.transforms.Compose
34
+ transforms:
35
+ - _partial_: true
36
+ _target_: sam3d_objects.data.dataset.tdfy.img_processing.pad_to_square_centered
37
+ - _target_: torchvision.transforms.Resize
38
+ size: 518
39
+ mask_transform:
40
+ _target_: torchvision.transforms.Compose
41
+ transforms:
42
+ - _partial_: true
43
+ _target_: sam3d_objects.data.dataset.tdfy.img_processing.pad_to_square_centered
44
+ - _target_: torchvision.transforms.Resize
45
+ interpolation: 0
46
+ size: 518
47
+ normalize_pointmap: true
48
+ pointmap_normalizer:
49
+ _target_: sam3d_objects.data.dataset.tdfy.img_and_mask_transforms.ObjectCentricSSI
50
+ allow_scale_and_shift_override: true
51
+ use_scene_scale: true
52
+ pointmap_transform:
53
+ _target_: torchvision.transforms.Compose
54
+ transforms:
55
+ - _partial_: true
56
+ _target_: sam3d_objects.data.dataset.tdfy.img_processing.pad_to_square_centered
57
+ - _target_: torchvision.transforms.Resize
58
+ interpolation: 0
59
+ size: 518
60
+ pose_decoder_name: ScaleShiftInvariant
61
+ depth_model:
62
+ _target_: sam3d_objects.pipeline.depth_models.moge.MoGe
63
+ model:
64
+ _target_: moge.model.v1.MoGeModel.from_pretrained
65
+ pretrained_model_name_or_path: Ruicheng/moge-vitl
66
+ slat_condition_input_mapping: []
67
+ slat_preprocessor:
68
+ _target_: sam3d_objects.data.dataset.tdfy.preprocessor.PreProcessor
69
+ img_transform:
70
+ _target_: torchvision.transforms.Compose
71
+ transforms:
72
+ - _target_: sam3d_objects.data.dataset.tdfy.img_processing.pad_to_square_centered
73
+ _partial_: true
74
+ - _target_: torchvision.transforms.Resize
75
+ size: 518
76
+ mask_transform:
77
+ _target_: torchvision.transforms.Compose
78
+ transforms:
79
+ - _target_: sam3d_objects.data.dataset.tdfy.img_processing.pad_to_square_centered
80
+ _partial_: true
81
+ - _target_: torchvision.transforms.Resize
82
+ size: 518
83
+ interpolation: 0
84
+ img_mask_joint_transform:
85
+ - _target_: sam3d_objects.data.dataset.tdfy.img_and_mask_transforms.crop_around_mask_with_padding
86
+ _partial_: true
87
+ box_size_factor: 1.2
88
+ padding_factor: 0.0
89
+ slat_mean:
90
+ - 0.12211431
91
+ - 0.37204156
92
+ - -1.26521907
93
+ - -2.05276058
94
+ - -3.10432536
95
+ - -0.11294304
96
+ - -0.85146744
97
+ - 0.45506954
98
+ slat_std:
99
+ - 2.37326008
100
+ - 2.13174402
101
+ - 2.2413953
102
+ - 2.30589401
103
+ - 2.1191894
104
+ - 1.8969511
105
+ - 2.41684989
106
+ - 2.08374642
checkpoints/slat_decoder_gs.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:785e9e4bfc8b27d316860e61a62655c5432d5e18ded4be9cdd488bcbbe7279bf
3
+ size 341486380
checkpoints/slat_decoder_gs.yaml ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.structured_latent_vae.decoder_gs.SLatGaussianDecoderTdfyWrapper
2
+ resolution: 64
3
+ model_channels: 768
4
+ latent_channels: 8
5
+ num_blocks: 12
6
+ num_heads: 12
7
+ mlp_ratio: 4
8
+ attn_mode: swin
9
+ window_size: 8
10
+ representation_config:
11
+ lr:
12
+ _xyz: 1.0
13
+ _features_dc: 1.0
14
+ _opacity: 1.0
15
+ _scaling: 1.0
16
+ _rotation: 0.1
17
+ perturb_offset: true
18
+ voxel_size: 1.5
19
+ num_gaussians: 32
20
+ 2d_filter_kernel_size: 0.1
21
+ 3d_filter_kernel_size: 0.0009
22
+ scaling_bias: 0.004
23
+ opacity_bias: 0.1
24
+ scaling_activation: softplus
25
+ use_fp16: true
checkpoints/slat_decoder_gs_4.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:301f87f722ebd13b4f64f74754572b716cedb6d10e86e0d2bea7beacc93438f1
3
+ size 340280249
checkpoints/slat_decoder_gs_4.yaml ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.structured_latent_vae.decoder_gs.SLatGaussianDecoderTdfyWrapper
2
+ resolution: 64
3
+ model_channels: 768
4
+ latent_channels: 8
5
+ num_blocks: 12
6
+ num_heads: 12
7
+ mlp_ratio: 4
8
+ attn_mode: swin
9
+ window_size: 8
10
+ representation_config:
11
+ lr:
12
+ _xyz: 1.0
13
+ _features_dc: 1.0
14
+ _opacity: 1.0
15
+ _scaling: 1.0
16
+ _rotation: 0.1
17
+ perturb_offset: true
18
+ voxel_size: 1.5
19
+ num_gaussians: 4
20
+ 2d_filter_kernel_size: 0.1
21
+ 3d_filter_kernel_size: 0.0009
22
+ scaling_bias: 0.004
23
+ opacity_bias: 0.1
24
+ scaling_activation: softplus
25
+ use_fp16: true
checkpoints/slat_decoder_mesh.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51f27a1c7a7892eff9576cf73bb888a255d4a184a76a9bfd9675f0d9ba464f5a
3
+ size 363718540
checkpoints/slat_decoder_mesh.yaml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.structured_latent_vae.decoder_mesh.SLatMeshDecoderTdfyWrapper
2
+ resolution: 64
3
+ model_channels: 768
4
+ latent_channels: 8
5
+ num_blocks: 12
6
+ num_heads: 12
7
+ mlp_ratio: 4
8
+ attn_mode: swin
9
+ window_size: 8
10
+ representation_config:
11
+ use_color: true
12
+ use_fp16: true
checkpoints/slat_generator.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee0c1601f3259f61cb22030bad839a3fe30a95b5f78bc9451babdbd8836fd06d
3
+ size 4906129160
checkpoints/slat_generator.yaml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ module:
2
+ condition_embedder:
3
+ backbone:
4
+ _target_: sam3d_objects.model.backbone.dit.embedder.embedder_fuser.EmbedderFuser
5
+ embedder_list:
6
+ - - _target_: sam3d_objects.model.backbone.dit.embedder.dino.Dino
7
+ dino_model: dinov2_vitl14_reg
8
+ input_size: 518
9
+ normalize_images: true
10
+ prenorm_features: true
11
+ - - - image
12
+ - cropped
13
+ - - rgb_image
14
+ - full
15
+ - - _target_: sam3d_objects.model.backbone.dit.embedder.dino.Dino
16
+ dino_model: dinov2_vitl14_reg
17
+ input_size: 518
18
+ normalize_images: true
19
+ prenorm_features: true
20
+ - - - mask
21
+ - cropped
22
+ - - rgb_image_mask
23
+ - full
24
+ projection_net_hidden_dim_multiplier: 4.0
25
+ use_pos_embedding: learned
26
+ generator:
27
+ backbone:
28
+ _target_: sam3d_objects.model.backbone.generator.flow_matching.model.FlowMatching
29
+ inference_steps: 12
30
+ reverse_fn:
31
+ _target_: sam3d_objects.model.backbone.generator.classifier_free_guidance.ClassifierFreeGuidance
32
+ backbone:
33
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.structured_latent_flow.SLatFlowModelTdfyWrapper
34
+ cond_channels: 1024
35
+ condition_embedder: null
36
+ force_zeros_cond: true
37
+ in_channels: 8
38
+ io_block_channels:
39
+ - 128
40
+ mlp_ratio: 4
41
+ model_channels: 1024
42
+ num_blocks: 24
43
+ num_heads: 16
44
+ num_io_res_blocks: 2
45
+ out_channels: 8
46
+ patch_size: 2
47
+ pe_mode: ape
48
+ qk_rms_norm: true
49
+ resolution: 64
50
+ use_fp16: true
51
+ p_unconditional: 0.0
52
+ strength: 0.0
53
+ unconditional_handling: add_flag
54
+ sigma_min: 0.0
55
+ time_scale: 1000.0
56
+ training_time_sampler_fn:
57
+ _partial_: true
58
+ _target_: sam3d_objects.model.backbone.generator.flow_matching.model.lognorm_sampler
59
+ mean: -1.0
60
+ std: 1.0
checkpoints/ss_decoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e87f8452dcb51399b96c6335cfc0ad17ec47ee87df9e1df17fae1a14a0619b7e
3
+ size 294691827
checkpoints/ss_generator.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f0fd0a7e288eadf7e51b6ead30f70aea0cd2ac4fe4160ee1deb412916bcfa0d
3
+ size 6689521984
checkpoints/ss_generator.yaml ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ module:
2
+ condition_embedder:
3
+ backbone:
4
+ _target_: sam3d_objects.model.backbone.dit.embedder.embedder_fuser.EmbedderFuser
5
+ drop_modalities_weight:
6
+ - - - pointmap
7
+ - rgb_pointmap
8
+ - 1.0
9
+ dropout_prob: 0.1
10
+ embedder_list:
11
+ - - _target_: sam3d_objects.model.backbone.dit.embedder.dino.Dino
12
+ dino_model: dinov2_vitl14_reg
13
+ input_size: 518
14
+ normalize_images: true
15
+ - - - image
16
+ - cropped
17
+ - - rgb_image
18
+ - full
19
+ - - _target_: sam3d_objects.model.backbone.dit.embedder.dino.Dino
20
+ dino_model: dinov2_vitl14_reg
21
+ input_size: 518
22
+ normalize_images: true
23
+ - - - mask
24
+ - cropped
25
+ - - rgb_image_mask
26
+ - full
27
+ - - _target_: sam3d_objects.model.backbone.dit.embedder.pointmap.PointPatchEmbed
28
+ embed_dim: 512
29
+ input_size: 256
30
+ patch_size: 8
31
+ remap_output: linear
32
+ - - - pointmap
33
+ - cropped
34
+ - - rgb_pointmap
35
+ - full
36
+ force_drop_modalities: null
37
+ freeze: true
38
+ projection_net_hidden_dim_multiplier: 4.0
39
+ use_pos_embedding: learned
40
+ generator:
41
+ backbone:
42
+ _target_: sam3d_objects.model.backbone.generator.shortcut.model.ShortCut
43
+ batch_mode: true
44
+ cfg_modalities:
45
+ - shape
46
+ inference_steps: 2
47
+ loss_weights:
48
+ 6drotation_normalized: 0.1
49
+ _target_: sam3d_objects.config.utils.make_dict
50
+ scale: 0.1
51
+ shape: 0
52
+ translation: 1.0
53
+ translation_scale: 0.0
54
+ ratio_cfg_samples_in_self_consistency_target: 0.25
55
+ rescale_t: 1
56
+ reverse_fn:
57
+ _target_: sam3d_objects.model.backbone.generator.classifier_free_guidance.ClassifierFreeGuidanceWithExternalUnconditionalProbability
58
+ backbone:
59
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mot_sparse_structure_flow.SparseStructureFlowTdfyWrapper
60
+ cond_channels: 1024
61
+ condition_embedder: null
62
+ force_zeros_cond: true
63
+ freeze_d_time_embedder: true
64
+ freeze_shared_parameters: true
65
+ in_channels: 8
66
+ is_shortcut_model: true
67
+ latent_mapping:
68
+ 6drotation_normalized:
69
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.Latent
70
+ in_channels: 6
71
+ model_channels: 1024
72
+ pos_embedder:
73
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.LearntPositionEmbedder
74
+ model_channels: 1024
75
+ token_len: 1
76
+ scale:
77
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.Latent
78
+ in_channels: 3
79
+ model_channels: 1024
80
+ pos_embedder:
81
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.LearntPositionEmbedder
82
+ model_channels: 1024
83
+ token_len: 1
84
+ shape:
85
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.Latent
86
+ in_channels: 8
87
+ model_channels: 1024
88
+ pos_embedder:
89
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.ShapePositionEmbedder
90
+ model_channels: 1024
91
+ patch_size: 1
92
+ resolution: 16
93
+ translation:
94
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.Latent
95
+ in_channels: 3
96
+ model_channels: 1024
97
+ pos_embedder:
98
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.LearntPositionEmbedder
99
+ model_channels: 1024
100
+ token_len: 1
101
+ translation_scale:
102
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.Latent
103
+ in_channels: 1
104
+ model_channels: 1024
105
+ pos_embedder:
106
+ _target_: sam3d_objects.model.backbone.tdfy_dit.models.mm_latent.LearntPositionEmbedder
107
+ model_channels: 1024
108
+ token_len: 1
109
+ latent_share_transformer:
110
+ 6drotation_normalized:
111
+ - 6drotation_normalized
112
+ - translation
113
+ - scale
114
+ - translation_scale
115
+ mlp_ratio: 4
116
+ model_channels: 1024
117
+ num_blocks: 24
118
+ num_heads: 16
119
+ out_channels: 8
120
+ patch_size: 1
121
+ pe_mode: ape
122
+ qk_rms_norm: true
123
+ resolution: 16
124
+ use_checkpoint: false
125
+ use_fp16: false
126
+ interval:
127
+ - 0
128
+ - 500
129
+ p_unconditional: 0.1
130
+ strength: 2.0
131
+ unconditional_handling: add_flag
132
+ self_consistency_cfg_strength: 2.0
133
+ self_consistency_prob: 0.25
134
+ shortcut_loss_weight: 1.0
135
+ sigma_min: 0.0
136
+ time_scale: 1000.0
137
+ training_time_sampler_fn:
138
+ _partial_: true
139
+ _target_: sam3d_objects.model.backbone.generator.flow_matching.model.lognorm_sampler
140
+ mean: -1.0
141
+ std: 1.0
moge/conversion_metadata/model.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ source_sha256: da96b09a0485a3c45a5aa455e67743c8b4efc4dd8437c1f2aa93c2b4303d957f
2
+ converter: mlx-spatial-restricted-torch-zip
3
+ tensor_count: 421
moge/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbdf89d1f7651b9213a507923b4f100fed87489bb7425fda7f7d430bcb31d8a9
3
+ size 1256740032