robotensor-attune 0.2.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,4826 @@
1
+ {
2
+ "action_proj.fc1.bias": {
3
+ "dtype": "F32",
4
+ "shape": [
5
+ 300
6
+ ]
7
+ },
8
+ "action_proj.fc1.weight": {
9
+ "dtype": "F32",
10
+ "shape": [
11
+ 300,
12
+ 300
13
+ ]
14
+ },
15
+ "action_proj.fc2.bias": {
16
+ "dtype": "F32",
17
+ "shape": [
18
+ 768
19
+ ]
20
+ },
21
+ "action_proj.fc2.weight": {
22
+ "dtype": "F32",
23
+ "shape": [
24
+ 768,
25
+ 300
26
+ ]
27
+ },
28
+ "decoder.layers.0.linear1.bias": {
29
+ "dtype": "F32",
30
+ "shape": [
31
+ 3072
32
+ ]
33
+ },
34
+ "decoder.layers.0.linear1.weight": {
35
+ "dtype": "F32",
36
+ "shape": [
37
+ 3072,
38
+ 768
39
+ ]
40
+ },
41
+ "decoder.layers.0.linear2.bias": {
42
+ "dtype": "F32",
43
+ "shape": [
44
+ 768
45
+ ]
46
+ },
47
+ "decoder.layers.0.linear2.weight": {
48
+ "dtype": "F32",
49
+ "shape": [
50
+ 768,
51
+ 3072
52
+ ]
53
+ },
54
+ "decoder.layers.0.multihead_attn.in_proj_bias": {
55
+ "dtype": "F32",
56
+ "shape": [
57
+ 2304
58
+ ]
59
+ },
60
+ "decoder.layers.0.multihead_attn.in_proj_weight": {
61
+ "dtype": "F32",
62
+ "shape": [
63
+ 2304,
64
+ 768
65
+ ]
66
+ },
67
+ "decoder.layers.0.multihead_attn.out_proj.bias": {
68
+ "dtype": "F32",
69
+ "shape": [
70
+ 768
71
+ ]
72
+ },
73
+ "decoder.layers.0.multihead_attn.out_proj.weight": {
74
+ "dtype": "F32",
75
+ "shape": [
76
+ 768,
77
+ 768
78
+ ]
79
+ },
80
+ "decoder.layers.0.norm1.bias": {
81
+ "dtype": "F32",
82
+ "shape": [
83
+ 768
84
+ ]
85
+ },
86
+ "decoder.layers.0.norm1.weight": {
87
+ "dtype": "F32",
88
+ "shape": [
89
+ 768
90
+ ]
91
+ },
92
+ "decoder.layers.0.norm2.bias": {
93
+ "dtype": "F32",
94
+ "shape": [
95
+ 768
96
+ ]
97
+ },
98
+ "decoder.layers.0.norm2.weight": {
99
+ "dtype": "F32",
100
+ "shape": [
101
+ 768
102
+ ]
103
+ },
104
+ "decoder.layers.0.norm3.bias": {
105
+ "dtype": "F32",
106
+ "shape": [
107
+ 768
108
+ ]
109
+ },
110
+ "decoder.layers.0.norm3.weight": {
111
+ "dtype": "F32",
112
+ "shape": [
113
+ 768
114
+ ]
115
+ },
116
+ "decoder.layers.0.self_attn.in_proj_bias": {
117
+ "dtype": "F32",
118
+ "shape": [
119
+ 2304
120
+ ]
121
+ },
122
+ "decoder.layers.0.self_attn.in_proj_weight": {
123
+ "dtype": "F32",
124
+ "shape": [
125
+ 2304,
126
+ 768
127
+ ]
128
+ },
129
+ "decoder.layers.0.self_attn.out_proj.bias": {
130
+ "dtype": "F32",
131
+ "shape": [
132
+ 768
133
+ ]
134
+ },
135
+ "decoder.layers.0.self_attn.out_proj.weight": {
136
+ "dtype": "F32",
137
+ "shape": [
138
+ 768,
139
+ 768
140
+ ]
141
+ },
142
+ "decoder.layers.1.linear1.bias": {
143
+ "dtype": "F32",
144
+ "shape": [
145
+ 3072
146
+ ]
147
+ },
148
+ "decoder.layers.1.linear1.weight": {
149
+ "dtype": "F32",
150
+ "shape": [
151
+ 3072,
152
+ 768
153
+ ]
154
+ },
155
+ "decoder.layers.1.linear2.bias": {
156
+ "dtype": "F32",
157
+ "shape": [
158
+ 768
159
+ ]
160
+ },
161
+ "decoder.layers.1.linear2.weight": {
162
+ "dtype": "F32",
163
+ "shape": [
164
+ 768,
165
+ 3072
166
+ ]
167
+ },
168
+ "decoder.layers.1.multihead_attn.in_proj_bias": {
169
+ "dtype": "F32",
170
+ "shape": [
171
+ 2304
172
+ ]
173
+ },
174
+ "decoder.layers.1.multihead_attn.in_proj_weight": {
175
+ "dtype": "F32",
176
+ "shape": [
177
+ 2304,
178
+ 768
179
+ ]
180
+ },
181
+ "decoder.layers.1.multihead_attn.out_proj.bias": {
182
+ "dtype": "F32",
183
+ "shape": [
184
+ 768
185
+ ]
186
+ },
187
+ "decoder.layers.1.multihead_attn.out_proj.weight": {
188
+ "dtype": "F32",
189
+ "shape": [
190
+ 768,
191
+ 768
192
+ ]
193
+ },
194
+ "decoder.layers.1.norm1.bias": {
195
+ "dtype": "F32",
196
+ "shape": [
197
+ 768
198
+ ]
199
+ },
200
+ "decoder.layers.1.norm1.weight": {
201
+ "dtype": "F32",
202
+ "shape": [
203
+ 768
204
+ ]
205
+ },
206
+ "decoder.layers.1.norm2.bias": {
207
+ "dtype": "F32",
208
+ "shape": [
209
+ 768
210
+ ]
211
+ },
212
+ "decoder.layers.1.norm2.weight": {
213
+ "dtype": "F32",
214
+ "shape": [
215
+ 768
216
+ ]
217
+ },
218
+ "decoder.layers.1.norm3.bias": {
219
+ "dtype": "F32",
220
+ "shape": [
221
+ 768
222
+ ]
223
+ },
224
+ "decoder.layers.1.norm3.weight": {
225
+ "dtype": "F32",
226
+ "shape": [
227
+ 768
228
+ ]
229
+ },
230
+ "decoder.layers.1.self_attn.in_proj_bias": {
231
+ "dtype": "F32",
232
+ "shape": [
233
+ 2304
234
+ ]
235
+ },
236
+ "decoder.layers.1.self_attn.in_proj_weight": {
237
+ "dtype": "F32",
238
+ "shape": [
239
+ 2304,
240
+ 768
241
+ ]
242
+ },
243
+ "decoder.layers.1.self_attn.out_proj.bias": {
244
+ "dtype": "F32",
245
+ "shape": [
246
+ 768
247
+ ]
248
+ },
249
+ "decoder.layers.1.self_attn.out_proj.weight": {
250
+ "dtype": "F32",
251
+ "shape": [
252
+ 768,
253
+ 768
254
+ ]
255
+ },
256
+ "decoder.layers.2.linear1.bias": {
257
+ "dtype": "F32",
258
+ "shape": [
259
+ 3072
260
+ ]
261
+ },
262
+ "decoder.layers.2.linear1.weight": {
263
+ "dtype": "F32",
264
+ "shape": [
265
+ 3072,
266
+ 768
267
+ ]
268
+ },
269
+ "decoder.layers.2.linear2.bias": {
270
+ "dtype": "F32",
271
+ "shape": [
272
+ 768
273
+ ]
274
+ },
275
+ "decoder.layers.2.linear2.weight": {
276
+ "dtype": "F32",
277
+ "shape": [
278
+ 768,
279
+ 3072
280
+ ]
281
+ },
282
+ "decoder.layers.2.multihead_attn.in_proj_bias": {
283
+ "dtype": "F32",
284
+ "shape": [
285
+ 2304
286
+ ]
287
+ },
288
+ "decoder.layers.2.multihead_attn.in_proj_weight": {
289
+ "dtype": "F32",
290
+ "shape": [
291
+ 2304,
292
+ 768
293
+ ]
294
+ },
295
+ "decoder.layers.2.multihead_attn.out_proj.bias": {
296
+ "dtype": "F32",
297
+ "shape": [
298
+ 768
299
+ ]
300
+ },
301
+ "decoder.layers.2.multihead_attn.out_proj.weight": {
302
+ "dtype": "F32",
303
+ "shape": [
304
+ 768,
305
+ 768
306
+ ]
307
+ },
308
+ "decoder.layers.2.norm1.bias": {
309
+ "dtype": "F32",
310
+ "shape": [
311
+ 768
312
+ ]
313
+ },
314
+ "decoder.layers.2.norm1.weight": {
315
+ "dtype": "F32",
316
+ "shape": [
317
+ 768
318
+ ]
319
+ },
320
+ "decoder.layers.2.norm2.bias": {
321
+ "dtype": "F32",
322
+ "shape": [
323
+ 768
324
+ ]
325
+ },
326
+ "decoder.layers.2.norm2.weight": {
327
+ "dtype": "F32",
328
+ "shape": [
329
+ 768
330
+ ]
331
+ },
332
+ "decoder.layers.2.norm3.bias": {
333
+ "dtype": "F32",
334
+ "shape": [
335
+ 768
336
+ ]
337
+ },
338
+ "decoder.layers.2.norm3.weight": {
339
+ "dtype": "F32",
340
+ "shape": [
341
+ 768
342
+ ]
343
+ },
344
+ "decoder.layers.2.self_attn.in_proj_bias": {
345
+ "dtype": "F32",
346
+ "shape": [
347
+ 2304
348
+ ]
349
+ },
350
+ "decoder.layers.2.self_attn.in_proj_weight": {
351
+ "dtype": "F32",
352
+ "shape": [
353
+ 2304,
354
+ 768
355
+ ]
356
+ },
357
+ "decoder.layers.2.self_attn.out_proj.bias": {
358
+ "dtype": "F32",
359
+ "shape": [
360
+ 768
361
+ ]
362
+ },
363
+ "decoder.layers.2.self_attn.out_proj.weight": {
364
+ "dtype": "F32",
365
+ "shape": [
366
+ 768,
367
+ 768
368
+ ]
369
+ },
370
+ "decoder.layers.3.linear1.bias": {
371
+ "dtype": "F32",
372
+ "shape": [
373
+ 3072
374
+ ]
375
+ },
376
+ "decoder.layers.3.linear1.weight": {
377
+ "dtype": "F32",
378
+ "shape": [
379
+ 3072,
380
+ 768
381
+ ]
382
+ },
383
+ "decoder.layers.3.linear2.bias": {
384
+ "dtype": "F32",
385
+ "shape": [
386
+ 768
387
+ ]
388
+ },
389
+ "decoder.layers.3.linear2.weight": {
390
+ "dtype": "F32",
391
+ "shape": [
392
+ 768,
393
+ 3072
394
+ ]
395
+ },
396
+ "decoder.layers.3.multihead_attn.in_proj_bias": {
397
+ "dtype": "F32",
398
+ "shape": [
399
+ 2304
400
+ ]
401
+ },
402
+ "decoder.layers.3.multihead_attn.in_proj_weight": {
403
+ "dtype": "F32",
404
+ "shape": [
405
+ 2304,
406
+ 768
407
+ ]
408
+ },
409
+ "decoder.layers.3.multihead_attn.out_proj.bias": {
410
+ "dtype": "F32",
411
+ "shape": [
412
+ 768
413
+ ]
414
+ },
415
+ "decoder.layers.3.multihead_attn.out_proj.weight": {
416
+ "dtype": "F32",
417
+ "shape": [
418
+ 768,
419
+ 768
420
+ ]
421
+ },
422
+ "decoder.layers.3.norm1.bias": {
423
+ "dtype": "F32",
424
+ "shape": [
425
+ 768
426
+ ]
427
+ },
428
+ "decoder.layers.3.norm1.weight": {
429
+ "dtype": "F32",
430
+ "shape": [
431
+ 768
432
+ ]
433
+ },
434
+ "decoder.layers.3.norm2.bias": {
435
+ "dtype": "F32",
436
+ "shape": [
437
+ 768
438
+ ]
439
+ },
440
+ "decoder.layers.3.norm2.weight": {
441
+ "dtype": "F32",
442
+ "shape": [
443
+ 768
444
+ ]
445
+ },
446
+ "decoder.layers.3.norm3.bias": {
447
+ "dtype": "F32",
448
+ "shape": [
449
+ 768
450
+ ]
451
+ },
452
+ "decoder.layers.3.norm3.weight": {
453
+ "dtype": "F32",
454
+ "shape": [
455
+ 768
456
+ ]
457
+ },
458
+ "decoder.layers.3.self_attn.in_proj_bias": {
459
+ "dtype": "F32",
460
+ "shape": [
461
+ 2304
462
+ ]
463
+ },
464
+ "decoder.layers.3.self_attn.in_proj_weight": {
465
+ "dtype": "F32",
466
+ "shape": [
467
+ 2304,
468
+ 768
469
+ ]
470
+ },
471
+ "decoder.layers.3.self_attn.out_proj.bias": {
472
+ "dtype": "F32",
473
+ "shape": [
474
+ 768
475
+ ]
476
+ },
477
+ "decoder.layers.3.self_attn.out_proj.weight": {
478
+ "dtype": "F32",
479
+ "shape": [
480
+ 768,
481
+ 768
482
+ ]
483
+ },
484
+ "decoder.layers.4.linear1.bias": {
485
+ "dtype": "F32",
486
+ "shape": [
487
+ 3072
488
+ ]
489
+ },
490
+ "decoder.layers.4.linear1.weight": {
491
+ "dtype": "F32",
492
+ "shape": [
493
+ 3072,
494
+ 768
495
+ ]
496
+ },
497
+ "decoder.layers.4.linear2.bias": {
498
+ "dtype": "F32",
499
+ "shape": [
500
+ 768
501
+ ]
502
+ },
503
+ "decoder.layers.4.linear2.weight": {
504
+ "dtype": "F32",
505
+ "shape": [
506
+ 768,
507
+ 3072
508
+ ]
509
+ },
510
+ "decoder.layers.4.multihead_attn.in_proj_bias": {
511
+ "dtype": "F32",
512
+ "shape": [
513
+ 2304
514
+ ]
515
+ },
516
+ "decoder.layers.4.multihead_attn.in_proj_weight": {
517
+ "dtype": "F32",
518
+ "shape": [
519
+ 2304,
520
+ 768
521
+ ]
522
+ },
523
+ "decoder.layers.4.multihead_attn.out_proj.bias": {
524
+ "dtype": "F32",
525
+ "shape": [
526
+ 768
527
+ ]
528
+ },
529
+ "decoder.layers.4.multihead_attn.out_proj.weight": {
530
+ "dtype": "F32",
531
+ "shape": [
532
+ 768,
533
+ 768
534
+ ]
535
+ },
536
+ "decoder.layers.4.norm1.bias": {
537
+ "dtype": "F32",
538
+ "shape": [
539
+ 768
540
+ ]
541
+ },
542
+ "decoder.layers.4.norm1.weight": {
543
+ "dtype": "F32",
544
+ "shape": [
545
+ 768
546
+ ]
547
+ },
548
+ "decoder.layers.4.norm2.bias": {
549
+ "dtype": "F32",
550
+ "shape": [
551
+ 768
552
+ ]
553
+ },
554
+ "decoder.layers.4.norm2.weight": {
555
+ "dtype": "F32",
556
+ "shape": [
557
+ 768
558
+ ]
559
+ },
560
+ "decoder.layers.4.norm3.bias": {
561
+ "dtype": "F32",
562
+ "shape": [
563
+ 768
564
+ ]
565
+ },
566
+ "decoder.layers.4.norm3.weight": {
567
+ "dtype": "F32",
568
+ "shape": [
569
+ 768
570
+ ]
571
+ },
572
+ "decoder.layers.4.self_attn.in_proj_bias": {
573
+ "dtype": "F32",
574
+ "shape": [
575
+ 2304
576
+ ]
577
+ },
578
+ "decoder.layers.4.self_attn.in_proj_weight": {
579
+ "dtype": "F32",
580
+ "shape": [
581
+ 2304,
582
+ 768
583
+ ]
584
+ },
585
+ "decoder.layers.4.self_attn.out_proj.bias": {
586
+ "dtype": "F32",
587
+ "shape": [
588
+ 768
589
+ ]
590
+ },
591
+ "decoder.layers.4.self_attn.out_proj.weight": {
592
+ "dtype": "F32",
593
+ "shape": [
594
+ 768,
595
+ 768
596
+ ]
597
+ },
598
+ "decoder.layers.5.linear1.bias": {
599
+ "dtype": "F32",
600
+ "shape": [
601
+ 3072
602
+ ]
603
+ },
604
+ "decoder.layers.5.linear1.weight": {
605
+ "dtype": "F32",
606
+ "shape": [
607
+ 3072,
608
+ 768
609
+ ]
610
+ },
611
+ "decoder.layers.5.linear2.bias": {
612
+ "dtype": "F32",
613
+ "shape": [
614
+ 768
615
+ ]
616
+ },
617
+ "decoder.layers.5.linear2.weight": {
618
+ "dtype": "F32",
619
+ "shape": [
620
+ 768,
621
+ 3072
622
+ ]
623
+ },
624
+ "decoder.layers.5.multihead_attn.in_proj_bias": {
625
+ "dtype": "F32",
626
+ "shape": [
627
+ 2304
628
+ ]
629
+ },
630
+ "decoder.layers.5.multihead_attn.in_proj_weight": {
631
+ "dtype": "F32",
632
+ "shape": [
633
+ 2304,
634
+ 768
635
+ ]
636
+ },
637
+ "decoder.layers.5.multihead_attn.out_proj.bias": {
638
+ "dtype": "F32",
639
+ "shape": [
640
+ 768
641
+ ]
642
+ },
643
+ "decoder.layers.5.multihead_attn.out_proj.weight": {
644
+ "dtype": "F32",
645
+ "shape": [
646
+ 768,
647
+ 768
648
+ ]
649
+ },
650
+ "decoder.layers.5.norm1.bias": {
651
+ "dtype": "F32",
652
+ "shape": [
653
+ 768
654
+ ]
655
+ },
656
+ "decoder.layers.5.norm1.weight": {
657
+ "dtype": "F32",
658
+ "shape": [
659
+ 768
660
+ ]
661
+ },
662
+ "decoder.layers.5.norm2.bias": {
663
+ "dtype": "F32",
664
+ "shape": [
665
+ 768
666
+ ]
667
+ },
668
+ "decoder.layers.5.norm2.weight": {
669
+ "dtype": "F32",
670
+ "shape": [
671
+ 768
672
+ ]
673
+ },
674
+ "decoder.layers.5.norm3.bias": {
675
+ "dtype": "F32",
676
+ "shape": [
677
+ 768
678
+ ]
679
+ },
680
+ "decoder.layers.5.norm3.weight": {
681
+ "dtype": "F32",
682
+ "shape": [
683
+ 768
684
+ ]
685
+ },
686
+ "decoder.layers.5.self_attn.in_proj_bias": {
687
+ "dtype": "F32",
688
+ "shape": [
689
+ 2304
690
+ ]
691
+ },
692
+ "decoder.layers.5.self_attn.in_proj_weight": {
693
+ "dtype": "F32",
694
+ "shape": [
695
+ 2304,
696
+ 768
697
+ ]
698
+ },
699
+ "decoder.layers.5.self_attn.out_proj.bias": {
700
+ "dtype": "F32",
701
+ "shape": [
702
+ 768
703
+ ]
704
+ },
705
+ "decoder.layers.5.self_attn.out_proj.weight": {
706
+ "dtype": "F32",
707
+ "shape": [
708
+ 768,
709
+ 768
710
+ ]
711
+ },
712
+ "norm.action.offset": {
713
+ "dtype": "F32",
714
+ "shape": [
715
+ 20
716
+ ]
717
+ },
718
+ "norm.action.scale": {
719
+ "dtype": "F32",
720
+ "shape": [
721
+ 20
722
+ ]
723
+ },
724
+ "norm.left_ee_ori.offset": {
725
+ "dtype": "F32",
726
+ "shape": [
727
+ 6
728
+ ]
729
+ },
730
+ "norm.left_ee_ori.scale": {
731
+ "dtype": "F32",
732
+ "shape": [
733
+ 6
734
+ ]
735
+ },
736
+ "norm.left_ee_pos.offset": {
737
+ "dtype": "F32",
738
+ "shape": [
739
+ 3
740
+ ]
741
+ },
742
+ "norm.left_ee_pos.scale": {
743
+ "dtype": "F32",
744
+ "shape": [
745
+ 3
746
+ ]
747
+ },
748
+ "norm.left_gripper_states.offset": {
749
+ "dtype": "F32",
750
+ "shape": [
751
+ 1
752
+ ]
753
+ },
754
+ "norm.left_gripper_states.scale": {
755
+ "dtype": "F32",
756
+ "shape": [
757
+ 1
758
+ ]
759
+ },
760
+ "norm.right_ee_ori.offset": {
761
+ "dtype": "F32",
762
+ "shape": [
763
+ 6
764
+ ]
765
+ },
766
+ "norm.right_ee_ori.scale": {
767
+ "dtype": "F32",
768
+ "shape": [
769
+ 6
770
+ ]
771
+ },
772
+ "norm.right_ee_pos.offset": {
773
+ "dtype": "F32",
774
+ "shape": [
775
+ 3
776
+ ]
777
+ },
778
+ "norm.right_ee_pos.scale": {
779
+ "dtype": "F32",
780
+ "shape": [
781
+ 3
782
+ ]
783
+ },
784
+ "norm.right_gripper_states.offset": {
785
+ "dtype": "F32",
786
+ "shape": [
787
+ 1
788
+ ]
789
+ },
790
+ "norm.right_gripper_states.scale": {
791
+ "dtype": "F32",
792
+ "shape": [
793
+ 1
794
+ ]
795
+ },
796
+ "obs_pos": {
797
+ "dtype": "F32",
798
+ "shape": [
799
+ 1,
800
+ 18,
801
+ 768
802
+ ]
803
+ },
804
+ "prompt_pool.kv.bias": {
805
+ "dtype": "F32",
806
+ "shape": [
807
+ 1536
808
+ ]
809
+ },
810
+ "prompt_pool.kv.weight": {
811
+ "dtype": "F32",
812
+ "shape": [
813
+ 1536,
814
+ 768
815
+ ]
816
+ },
817
+ "prompt_pool.latent": {
818
+ "dtype": "F32",
819
+ "shape": [
820
+ 1,
821
+ 1,
822
+ 768
823
+ ]
824
+ },
825
+ "prompt_pool.mlp.fc1.bias": {
826
+ "dtype": "F32",
827
+ "shape": [
828
+ 3072
829
+ ]
830
+ },
831
+ "prompt_pool.mlp.fc1.weight": {
832
+ "dtype": "F32",
833
+ "shape": [
834
+ 3072,
835
+ 768
836
+ ]
837
+ },
838
+ "prompt_pool.mlp.fc2.bias": {
839
+ "dtype": "F32",
840
+ "shape": [
841
+ 768
842
+ ]
843
+ },
844
+ "prompt_pool.mlp.fc2.weight": {
845
+ "dtype": "F32",
846
+ "shape": [
847
+ 768,
848
+ 3072
849
+ ]
850
+ },
851
+ "prompt_pool.proj.bias": {
852
+ "dtype": "F32",
853
+ "shape": [
854
+ 768
855
+ ]
856
+ },
857
+ "prompt_pool.proj.weight": {
858
+ "dtype": "F32",
859
+ "shape": [
860
+ 768,
861
+ 768
862
+ ]
863
+ },
864
+ "prompt_pool.q.bias": {
865
+ "dtype": "F32",
866
+ "shape": [
867
+ 768
868
+ ]
869
+ },
870
+ "prompt_pool.q.weight": {
871
+ "dtype": "F32",
872
+ "shape": [
873
+ 768,
874
+ 768
875
+ ]
876
+ },
877
+ "prompt_pos": {
878
+ "dtype": "F32",
879
+ "shape": [
880
+ 1,
881
+ 67,
882
+ 768
883
+ ]
884
+ },
885
+ "proprio.left_ee_ori.bias": {
886
+ "dtype": "F32",
887
+ "shape": [
888
+ 768
889
+ ]
890
+ },
891
+ "proprio.left_ee_ori.weight": {
892
+ "dtype": "F32",
893
+ "shape": [
894
+ 768,
895
+ 6
896
+ ]
897
+ },
898
+ "proprio.left_ee_pos.bias": {
899
+ "dtype": "F32",
900
+ "shape": [
901
+ 768
902
+ ]
903
+ },
904
+ "proprio.left_ee_pos.weight": {
905
+ "dtype": "F32",
906
+ "shape": [
907
+ 768,
908
+ 3
909
+ ]
910
+ },
911
+ "proprio.left_gripper_states.bias": {
912
+ "dtype": "F32",
913
+ "shape": [
914
+ 768
915
+ ]
916
+ },
917
+ "proprio.left_gripper_states.weight": {
918
+ "dtype": "F32",
919
+ "shape": [
920
+ 768,
921
+ 1
922
+ ]
923
+ },
924
+ "proprio.right_ee_ori.bias": {
925
+ "dtype": "F32",
926
+ "shape": [
927
+ 768
928
+ ]
929
+ },
930
+ "proprio.right_ee_ori.weight": {
931
+ "dtype": "F32",
932
+ "shape": [
933
+ 768,
934
+ 6
935
+ ]
936
+ },
937
+ "proprio.right_ee_pos.bias": {
938
+ "dtype": "F32",
939
+ "shape": [
940
+ 768
941
+ ]
942
+ },
943
+ "proprio.right_ee_pos.weight": {
944
+ "dtype": "F32",
945
+ "shape": [
946
+ 768,
947
+ 3
948
+ ]
949
+ },
950
+ "proprio.right_gripper_states.bias": {
951
+ "dtype": "F32",
952
+ "shape": [
953
+ 768
954
+ ]
955
+ },
956
+ "proprio.right_gripper_states.weight": {
957
+ "dtype": "F32",
958
+ "shape": [
959
+ 768,
960
+ 1
961
+ ]
962
+ },
963
+ "sinks": {
964
+ "dtype": "F32",
965
+ "shape": [
966
+ 1,
967
+ 4,
968
+ 768
969
+ ]
970
+ },
971
+ "unet.diffusion_step_encoder.1.bias": {
972
+ "dtype": "F32",
973
+ "shape": [
974
+ 512
975
+ ]
976
+ },
977
+ "unet.diffusion_step_encoder.1.weight": {
978
+ "dtype": "F32",
979
+ "shape": [
980
+ 512,
981
+ 128
982
+ ]
983
+ },
984
+ "unet.diffusion_step_encoder.3.bias": {
985
+ "dtype": "F32",
986
+ "shape": [
987
+ 128
988
+ ]
989
+ },
990
+ "unet.diffusion_step_encoder.3.weight": {
991
+ "dtype": "F32",
992
+ "shape": [
993
+ 128,
994
+ 512
995
+ ]
996
+ },
997
+ "unet.down_modules.0.0.blocks.0.block.0.bias": {
998
+ "dtype": "F32",
999
+ "shape": [
1000
+ 256
1001
+ ]
1002
+ },
1003
+ "unet.down_modules.0.0.blocks.0.block.0.weight": {
1004
+ "dtype": "F32",
1005
+ "shape": [
1006
+ 256,
1007
+ 20,
1008
+ 5
1009
+ ]
1010
+ },
1011
+ "unet.down_modules.0.0.blocks.0.block.1.bias": {
1012
+ "dtype": "F32",
1013
+ "shape": [
1014
+ 256
1015
+ ]
1016
+ },
1017
+ "unet.down_modules.0.0.blocks.0.block.1.weight": {
1018
+ "dtype": "F32",
1019
+ "shape": [
1020
+ 256
1021
+ ]
1022
+ },
1023
+ "unet.down_modules.0.0.blocks.1.block.0.bias": {
1024
+ "dtype": "F32",
1025
+ "shape": [
1026
+ 256
1027
+ ]
1028
+ },
1029
+ "unet.down_modules.0.0.blocks.1.block.0.weight": {
1030
+ "dtype": "F32",
1031
+ "shape": [
1032
+ 256,
1033
+ 256,
1034
+ 5
1035
+ ]
1036
+ },
1037
+ "unet.down_modules.0.0.blocks.1.block.1.bias": {
1038
+ "dtype": "F32",
1039
+ "shape": [
1040
+ 256
1041
+ ]
1042
+ },
1043
+ "unet.down_modules.0.0.blocks.1.block.1.weight": {
1044
+ "dtype": "F32",
1045
+ "shape": [
1046
+ 256
1047
+ ]
1048
+ },
1049
+ "unet.down_modules.0.0.cond_encoder.1.bias": {
1050
+ "dtype": "F32",
1051
+ "shape": [
1052
+ 512
1053
+ ]
1054
+ },
1055
+ "unet.down_modules.0.0.cond_encoder.1.weight": {
1056
+ "dtype": "F32",
1057
+ "shape": [
1058
+ 512,
1059
+ 27776
1060
+ ]
1061
+ },
1062
+ "unet.down_modules.0.0.residual_conv.bias": {
1063
+ "dtype": "F32",
1064
+ "shape": [
1065
+ 256
1066
+ ]
1067
+ },
1068
+ "unet.down_modules.0.0.residual_conv.weight": {
1069
+ "dtype": "F32",
1070
+ "shape": [
1071
+ 256,
1072
+ 20,
1073
+ 1
1074
+ ]
1075
+ },
1076
+ "unet.down_modules.0.1.blocks.0.block.0.bias": {
1077
+ "dtype": "F32",
1078
+ "shape": [
1079
+ 256
1080
+ ]
1081
+ },
1082
+ "unet.down_modules.0.1.blocks.0.block.0.weight": {
1083
+ "dtype": "F32",
1084
+ "shape": [
1085
+ 256,
1086
+ 256,
1087
+ 5
1088
+ ]
1089
+ },
1090
+ "unet.down_modules.0.1.blocks.0.block.1.bias": {
1091
+ "dtype": "F32",
1092
+ "shape": [
1093
+ 256
1094
+ ]
1095
+ },
1096
+ "unet.down_modules.0.1.blocks.0.block.1.weight": {
1097
+ "dtype": "F32",
1098
+ "shape": [
1099
+ 256
1100
+ ]
1101
+ },
1102
+ "unet.down_modules.0.1.blocks.1.block.0.bias": {
1103
+ "dtype": "F32",
1104
+ "shape": [
1105
+ 256
1106
+ ]
1107
+ },
1108
+ "unet.down_modules.0.1.blocks.1.block.0.weight": {
1109
+ "dtype": "F32",
1110
+ "shape": [
1111
+ 256,
1112
+ 256,
1113
+ 5
1114
+ ]
1115
+ },
1116
+ "unet.down_modules.0.1.blocks.1.block.1.bias": {
1117
+ "dtype": "F32",
1118
+ "shape": [
1119
+ 256
1120
+ ]
1121
+ },
1122
+ "unet.down_modules.0.1.blocks.1.block.1.weight": {
1123
+ "dtype": "F32",
1124
+ "shape": [
1125
+ 256
1126
+ ]
1127
+ },
1128
+ "unet.down_modules.0.1.cond_encoder.1.bias": {
1129
+ "dtype": "F32",
1130
+ "shape": [
1131
+ 512
1132
+ ]
1133
+ },
1134
+ "unet.down_modules.0.1.cond_encoder.1.weight": {
1135
+ "dtype": "F32",
1136
+ "shape": [
1137
+ 512,
1138
+ 27776
1139
+ ]
1140
+ },
1141
+ "unet.down_modules.0.2.bias": {
1142
+ "dtype": "F32",
1143
+ "shape": [
1144
+ 256
1145
+ ]
1146
+ },
1147
+ "unet.down_modules.0.2.weight": {
1148
+ "dtype": "F32",
1149
+ "shape": [
1150
+ 256,
1151
+ 256,
1152
+ 3
1153
+ ]
1154
+ },
1155
+ "unet.down_modules.1.0.blocks.0.block.0.bias": {
1156
+ "dtype": "F32",
1157
+ "shape": [
1158
+ 512
1159
+ ]
1160
+ },
1161
+ "unet.down_modules.1.0.blocks.0.block.0.weight": {
1162
+ "dtype": "F32",
1163
+ "shape": [
1164
+ 512,
1165
+ 256,
1166
+ 5
1167
+ ]
1168
+ },
1169
+ "unet.down_modules.1.0.blocks.0.block.1.bias": {
1170
+ "dtype": "F32",
1171
+ "shape": [
1172
+ 512
1173
+ ]
1174
+ },
1175
+ "unet.down_modules.1.0.blocks.0.block.1.weight": {
1176
+ "dtype": "F32",
1177
+ "shape": [
1178
+ 512
1179
+ ]
1180
+ },
1181
+ "unet.down_modules.1.0.blocks.1.block.0.bias": {
1182
+ "dtype": "F32",
1183
+ "shape": [
1184
+ 512
1185
+ ]
1186
+ },
1187
+ "unet.down_modules.1.0.blocks.1.block.0.weight": {
1188
+ "dtype": "F32",
1189
+ "shape": [
1190
+ 512,
1191
+ 512,
1192
+ 5
1193
+ ]
1194
+ },
1195
+ "unet.down_modules.1.0.blocks.1.block.1.bias": {
1196
+ "dtype": "F32",
1197
+ "shape": [
1198
+ 512
1199
+ ]
1200
+ },
1201
+ "unet.down_modules.1.0.blocks.1.block.1.weight": {
1202
+ "dtype": "F32",
1203
+ "shape": [
1204
+ 512
1205
+ ]
1206
+ },
1207
+ "unet.down_modules.1.0.cond_encoder.1.bias": {
1208
+ "dtype": "F32",
1209
+ "shape": [
1210
+ 1024
1211
+ ]
1212
+ },
1213
+ "unet.down_modules.1.0.cond_encoder.1.weight": {
1214
+ "dtype": "F32",
1215
+ "shape": [
1216
+ 1024,
1217
+ 27776
1218
+ ]
1219
+ },
1220
+ "unet.down_modules.1.0.residual_conv.bias": {
1221
+ "dtype": "F32",
1222
+ "shape": [
1223
+ 512
1224
+ ]
1225
+ },
1226
+ "unet.down_modules.1.0.residual_conv.weight": {
1227
+ "dtype": "F32",
1228
+ "shape": [
1229
+ 512,
1230
+ 256,
1231
+ 1
1232
+ ]
1233
+ },
1234
+ "unet.down_modules.1.1.blocks.0.block.0.bias": {
1235
+ "dtype": "F32",
1236
+ "shape": [
1237
+ 512
1238
+ ]
1239
+ },
1240
+ "unet.down_modules.1.1.blocks.0.block.0.weight": {
1241
+ "dtype": "F32",
1242
+ "shape": [
1243
+ 512,
1244
+ 512,
1245
+ 5
1246
+ ]
1247
+ },
1248
+ "unet.down_modules.1.1.blocks.0.block.1.bias": {
1249
+ "dtype": "F32",
1250
+ "shape": [
1251
+ 512
1252
+ ]
1253
+ },
1254
+ "unet.down_modules.1.1.blocks.0.block.1.weight": {
1255
+ "dtype": "F32",
1256
+ "shape": [
1257
+ 512
1258
+ ]
1259
+ },
1260
+ "unet.down_modules.1.1.blocks.1.block.0.bias": {
1261
+ "dtype": "F32",
1262
+ "shape": [
1263
+ 512
1264
+ ]
1265
+ },
1266
+ "unet.down_modules.1.1.blocks.1.block.0.weight": {
1267
+ "dtype": "F32",
1268
+ "shape": [
1269
+ 512,
1270
+ 512,
1271
+ 5
1272
+ ]
1273
+ },
1274
+ "unet.down_modules.1.1.blocks.1.block.1.bias": {
1275
+ "dtype": "F32",
1276
+ "shape": [
1277
+ 512
1278
+ ]
1279
+ },
1280
+ "unet.down_modules.1.1.blocks.1.block.1.weight": {
1281
+ "dtype": "F32",
1282
+ "shape": [
1283
+ 512
1284
+ ]
1285
+ },
1286
+ "unet.down_modules.1.1.cond_encoder.1.bias": {
1287
+ "dtype": "F32",
1288
+ "shape": [
1289
+ 1024
1290
+ ]
1291
+ },
1292
+ "unet.down_modules.1.1.cond_encoder.1.weight": {
1293
+ "dtype": "F32",
1294
+ "shape": [
1295
+ 1024,
1296
+ 27776
1297
+ ]
1298
+ },
1299
+ "unet.down_modules.1.2.bias": {
1300
+ "dtype": "F32",
1301
+ "shape": [
1302
+ 512
1303
+ ]
1304
+ },
1305
+ "unet.down_modules.1.2.weight": {
1306
+ "dtype": "F32",
1307
+ "shape": [
1308
+ 512,
1309
+ 512,
1310
+ 3
1311
+ ]
1312
+ },
1313
+ "unet.down_modules.2.0.blocks.0.block.0.bias": {
1314
+ "dtype": "F32",
1315
+ "shape": [
1316
+ 1024
1317
+ ]
1318
+ },
1319
+ "unet.down_modules.2.0.blocks.0.block.0.weight": {
1320
+ "dtype": "F32",
1321
+ "shape": [
1322
+ 1024,
1323
+ 512,
1324
+ 5
1325
+ ]
1326
+ },
1327
+ "unet.down_modules.2.0.blocks.0.block.1.bias": {
1328
+ "dtype": "F32",
1329
+ "shape": [
1330
+ 1024
1331
+ ]
1332
+ },
1333
+ "unet.down_modules.2.0.blocks.0.block.1.weight": {
1334
+ "dtype": "F32",
1335
+ "shape": [
1336
+ 1024
1337
+ ]
1338
+ },
1339
+ "unet.down_modules.2.0.blocks.1.block.0.bias": {
1340
+ "dtype": "F32",
1341
+ "shape": [
1342
+ 1024
1343
+ ]
1344
+ },
1345
+ "unet.down_modules.2.0.blocks.1.block.0.weight": {
1346
+ "dtype": "F32",
1347
+ "shape": [
1348
+ 1024,
1349
+ 1024,
1350
+ 5
1351
+ ]
1352
+ },
1353
+ "unet.down_modules.2.0.blocks.1.block.1.bias": {
1354
+ "dtype": "F32",
1355
+ "shape": [
1356
+ 1024
1357
+ ]
1358
+ },
1359
+ "unet.down_modules.2.0.blocks.1.block.1.weight": {
1360
+ "dtype": "F32",
1361
+ "shape": [
1362
+ 1024
1363
+ ]
1364
+ },
1365
+ "unet.down_modules.2.0.cond_encoder.1.bias": {
1366
+ "dtype": "F32",
1367
+ "shape": [
1368
+ 2048
1369
+ ]
1370
+ },
1371
+ "unet.down_modules.2.0.cond_encoder.1.weight": {
1372
+ "dtype": "F32",
1373
+ "shape": [
1374
+ 2048,
1375
+ 27776
1376
+ ]
1377
+ },
1378
+ "unet.down_modules.2.0.residual_conv.bias": {
1379
+ "dtype": "F32",
1380
+ "shape": [
1381
+ 1024
1382
+ ]
1383
+ },
1384
+ "unet.down_modules.2.0.residual_conv.weight": {
1385
+ "dtype": "F32",
1386
+ "shape": [
1387
+ 1024,
1388
+ 512,
1389
+ 1
1390
+ ]
1391
+ },
1392
+ "unet.down_modules.2.1.blocks.0.block.0.bias": {
1393
+ "dtype": "F32",
1394
+ "shape": [
1395
+ 1024
1396
+ ]
1397
+ },
1398
+ "unet.down_modules.2.1.blocks.0.block.0.weight": {
1399
+ "dtype": "F32",
1400
+ "shape": [
1401
+ 1024,
1402
+ 1024,
1403
+ 5
1404
+ ]
1405
+ },
1406
+ "unet.down_modules.2.1.blocks.0.block.1.bias": {
1407
+ "dtype": "F32",
1408
+ "shape": [
1409
+ 1024
1410
+ ]
1411
+ },
1412
+ "unet.down_modules.2.1.blocks.0.block.1.weight": {
1413
+ "dtype": "F32",
1414
+ "shape": [
1415
+ 1024
1416
+ ]
1417
+ },
1418
+ "unet.down_modules.2.1.blocks.1.block.0.bias": {
1419
+ "dtype": "F32",
1420
+ "shape": [
1421
+ 1024
1422
+ ]
1423
+ },
1424
+ "unet.down_modules.2.1.blocks.1.block.0.weight": {
1425
+ "dtype": "F32",
1426
+ "shape": [
1427
+ 1024,
1428
+ 1024,
1429
+ 5
1430
+ ]
1431
+ },
1432
+ "unet.down_modules.2.1.blocks.1.block.1.bias": {
1433
+ "dtype": "F32",
1434
+ "shape": [
1435
+ 1024
1436
+ ]
1437
+ },
1438
+ "unet.down_modules.2.1.blocks.1.block.1.weight": {
1439
+ "dtype": "F32",
1440
+ "shape": [
1441
+ 1024
1442
+ ]
1443
+ },
1444
+ "unet.down_modules.2.1.cond_encoder.1.bias": {
1445
+ "dtype": "F32",
1446
+ "shape": [
1447
+ 2048
1448
+ ]
1449
+ },
1450
+ "unet.down_modules.2.1.cond_encoder.1.weight": {
1451
+ "dtype": "F32",
1452
+ "shape": [
1453
+ 2048,
1454
+ 27776
1455
+ ]
1456
+ },
1457
+ "unet.final_conv.0.block.0.bias": {
1458
+ "dtype": "F32",
1459
+ "shape": [
1460
+ 256
1461
+ ]
1462
+ },
1463
+ "unet.final_conv.0.block.0.weight": {
1464
+ "dtype": "F32",
1465
+ "shape": [
1466
+ 256,
1467
+ 256,
1468
+ 5
1469
+ ]
1470
+ },
1471
+ "unet.final_conv.0.block.1.bias": {
1472
+ "dtype": "F32",
1473
+ "shape": [
1474
+ 256
1475
+ ]
1476
+ },
1477
+ "unet.final_conv.0.block.1.weight": {
1478
+ "dtype": "F32",
1479
+ "shape": [
1480
+ 256
1481
+ ]
1482
+ },
1483
+ "unet.final_conv.1.bias": {
1484
+ "dtype": "F32",
1485
+ "shape": [
1486
+ 20
1487
+ ]
1488
+ },
1489
+ "unet.final_conv.1.weight": {
1490
+ "dtype": "F32",
1491
+ "shape": [
1492
+ 20,
1493
+ 256,
1494
+ 1
1495
+ ]
1496
+ },
1497
+ "unet.mid_modules.0.blocks.0.block.0.bias": {
1498
+ "dtype": "F32",
1499
+ "shape": [
1500
+ 1024
1501
+ ]
1502
+ },
1503
+ "unet.mid_modules.0.blocks.0.block.0.weight": {
1504
+ "dtype": "F32",
1505
+ "shape": [
1506
+ 1024,
1507
+ 1024,
1508
+ 5
1509
+ ]
1510
+ },
1511
+ "unet.mid_modules.0.blocks.0.block.1.bias": {
1512
+ "dtype": "F32",
1513
+ "shape": [
1514
+ 1024
1515
+ ]
1516
+ },
1517
+ "unet.mid_modules.0.blocks.0.block.1.weight": {
1518
+ "dtype": "F32",
1519
+ "shape": [
1520
+ 1024
1521
+ ]
1522
+ },
1523
+ "unet.mid_modules.0.blocks.1.block.0.bias": {
1524
+ "dtype": "F32",
1525
+ "shape": [
1526
+ 1024
1527
+ ]
1528
+ },
1529
+ "unet.mid_modules.0.blocks.1.block.0.weight": {
1530
+ "dtype": "F32",
1531
+ "shape": [
1532
+ 1024,
1533
+ 1024,
1534
+ 5
1535
+ ]
1536
+ },
1537
+ "unet.mid_modules.0.blocks.1.block.1.bias": {
1538
+ "dtype": "F32",
1539
+ "shape": [
1540
+ 1024
1541
+ ]
1542
+ },
1543
+ "unet.mid_modules.0.blocks.1.block.1.weight": {
1544
+ "dtype": "F32",
1545
+ "shape": [
1546
+ 1024
1547
+ ]
1548
+ },
1549
+ "unet.mid_modules.0.cond_encoder.1.bias": {
1550
+ "dtype": "F32",
1551
+ "shape": [
1552
+ 2048
1553
+ ]
1554
+ },
1555
+ "unet.mid_modules.0.cond_encoder.1.weight": {
1556
+ "dtype": "F32",
1557
+ "shape": [
1558
+ 2048,
1559
+ 27776
1560
+ ]
1561
+ },
1562
+ "unet.mid_modules.1.blocks.0.block.0.bias": {
1563
+ "dtype": "F32",
1564
+ "shape": [
1565
+ 1024
1566
+ ]
1567
+ },
1568
+ "unet.mid_modules.1.blocks.0.block.0.weight": {
1569
+ "dtype": "F32",
1570
+ "shape": [
1571
+ 1024,
1572
+ 1024,
1573
+ 5
1574
+ ]
1575
+ },
1576
+ "unet.mid_modules.1.blocks.0.block.1.bias": {
1577
+ "dtype": "F32",
1578
+ "shape": [
1579
+ 1024
1580
+ ]
1581
+ },
1582
+ "unet.mid_modules.1.blocks.0.block.1.weight": {
1583
+ "dtype": "F32",
1584
+ "shape": [
1585
+ 1024
1586
+ ]
1587
+ },
1588
+ "unet.mid_modules.1.blocks.1.block.0.bias": {
1589
+ "dtype": "F32",
1590
+ "shape": [
1591
+ 1024
1592
+ ]
1593
+ },
1594
+ "unet.mid_modules.1.blocks.1.block.0.weight": {
1595
+ "dtype": "F32",
1596
+ "shape": [
1597
+ 1024,
1598
+ 1024,
1599
+ 5
1600
+ ]
1601
+ },
1602
+ "unet.mid_modules.1.blocks.1.block.1.bias": {
1603
+ "dtype": "F32",
1604
+ "shape": [
1605
+ 1024
1606
+ ]
1607
+ },
1608
+ "unet.mid_modules.1.blocks.1.block.1.weight": {
1609
+ "dtype": "F32",
1610
+ "shape": [
1611
+ 1024
1612
+ ]
1613
+ },
1614
+ "unet.mid_modules.1.cond_encoder.1.bias": {
1615
+ "dtype": "F32",
1616
+ "shape": [
1617
+ 2048
1618
+ ]
1619
+ },
1620
+ "unet.mid_modules.1.cond_encoder.1.weight": {
1621
+ "dtype": "F32",
1622
+ "shape": [
1623
+ 2048,
1624
+ 27776
1625
+ ]
1626
+ },
1627
+ "unet.up_modules.0.0.blocks.0.block.0.bias": {
1628
+ "dtype": "F32",
1629
+ "shape": [
1630
+ 512
1631
+ ]
1632
+ },
1633
+ "unet.up_modules.0.0.blocks.0.block.0.weight": {
1634
+ "dtype": "F32",
1635
+ "shape": [
1636
+ 512,
1637
+ 2048,
1638
+ 5
1639
+ ]
1640
+ },
1641
+ "unet.up_modules.0.0.blocks.0.block.1.bias": {
1642
+ "dtype": "F32",
1643
+ "shape": [
1644
+ 512
1645
+ ]
1646
+ },
1647
+ "unet.up_modules.0.0.blocks.0.block.1.weight": {
1648
+ "dtype": "F32",
1649
+ "shape": [
1650
+ 512
1651
+ ]
1652
+ },
1653
+ "unet.up_modules.0.0.blocks.1.block.0.bias": {
1654
+ "dtype": "F32",
1655
+ "shape": [
1656
+ 512
1657
+ ]
1658
+ },
1659
+ "unet.up_modules.0.0.blocks.1.block.0.weight": {
1660
+ "dtype": "F32",
1661
+ "shape": [
1662
+ 512,
1663
+ 512,
1664
+ 5
1665
+ ]
1666
+ },
1667
+ "unet.up_modules.0.0.blocks.1.block.1.bias": {
1668
+ "dtype": "F32",
1669
+ "shape": [
1670
+ 512
1671
+ ]
1672
+ },
1673
+ "unet.up_modules.0.0.blocks.1.block.1.weight": {
1674
+ "dtype": "F32",
1675
+ "shape": [
1676
+ 512
1677
+ ]
1678
+ },
1679
+ "unet.up_modules.0.0.cond_encoder.1.bias": {
1680
+ "dtype": "F32",
1681
+ "shape": [
1682
+ 1024
1683
+ ]
1684
+ },
1685
+ "unet.up_modules.0.0.cond_encoder.1.weight": {
1686
+ "dtype": "F32",
1687
+ "shape": [
1688
+ 1024,
1689
+ 27776
1690
+ ]
1691
+ },
1692
+ "unet.up_modules.0.0.residual_conv.bias": {
1693
+ "dtype": "F32",
1694
+ "shape": [
1695
+ 512
1696
+ ]
1697
+ },
1698
+ "unet.up_modules.0.0.residual_conv.weight": {
1699
+ "dtype": "F32",
1700
+ "shape": [
1701
+ 512,
1702
+ 2048,
1703
+ 1
1704
+ ]
1705
+ },
1706
+ "unet.up_modules.0.1.blocks.0.block.0.bias": {
1707
+ "dtype": "F32",
1708
+ "shape": [
1709
+ 512
1710
+ ]
1711
+ },
1712
+ "unet.up_modules.0.1.blocks.0.block.0.weight": {
1713
+ "dtype": "F32",
1714
+ "shape": [
1715
+ 512,
1716
+ 512,
1717
+ 5
1718
+ ]
1719
+ },
1720
+ "unet.up_modules.0.1.blocks.0.block.1.bias": {
1721
+ "dtype": "F32",
1722
+ "shape": [
1723
+ 512
1724
+ ]
1725
+ },
1726
+ "unet.up_modules.0.1.blocks.0.block.1.weight": {
1727
+ "dtype": "F32",
1728
+ "shape": [
1729
+ 512
1730
+ ]
1731
+ },
1732
+ "unet.up_modules.0.1.blocks.1.block.0.bias": {
1733
+ "dtype": "F32",
1734
+ "shape": [
1735
+ 512
1736
+ ]
1737
+ },
1738
+ "unet.up_modules.0.1.blocks.1.block.0.weight": {
1739
+ "dtype": "F32",
1740
+ "shape": [
1741
+ 512,
1742
+ 512,
1743
+ 5
1744
+ ]
1745
+ },
1746
+ "unet.up_modules.0.1.blocks.1.block.1.bias": {
1747
+ "dtype": "F32",
1748
+ "shape": [
1749
+ 512
1750
+ ]
1751
+ },
1752
+ "unet.up_modules.0.1.blocks.1.block.1.weight": {
1753
+ "dtype": "F32",
1754
+ "shape": [
1755
+ 512
1756
+ ]
1757
+ },
1758
+ "unet.up_modules.0.1.cond_encoder.1.bias": {
1759
+ "dtype": "F32",
1760
+ "shape": [
1761
+ 1024
1762
+ ]
1763
+ },
1764
+ "unet.up_modules.0.1.cond_encoder.1.weight": {
1765
+ "dtype": "F32",
1766
+ "shape": [
1767
+ 1024,
1768
+ 27776
1769
+ ]
1770
+ },
1771
+ "unet.up_modules.0.2.bias": {
1772
+ "dtype": "F32",
1773
+ "shape": [
1774
+ 512
1775
+ ]
1776
+ },
1777
+ "unet.up_modules.0.2.weight": {
1778
+ "dtype": "F32",
1779
+ "shape": [
1780
+ 512,
1781
+ 512,
1782
+ 4
1783
+ ]
1784
+ },
1785
+ "unet.up_modules.1.0.blocks.0.block.0.bias": {
1786
+ "dtype": "F32",
1787
+ "shape": [
1788
+ 256
1789
+ ]
1790
+ },
1791
+ "unet.up_modules.1.0.blocks.0.block.0.weight": {
1792
+ "dtype": "F32",
1793
+ "shape": [
1794
+ 256,
1795
+ 1024,
1796
+ 5
1797
+ ]
1798
+ },
1799
+ "unet.up_modules.1.0.blocks.0.block.1.bias": {
1800
+ "dtype": "F32",
1801
+ "shape": [
1802
+ 256
1803
+ ]
1804
+ },
1805
+ "unet.up_modules.1.0.blocks.0.block.1.weight": {
1806
+ "dtype": "F32",
1807
+ "shape": [
1808
+ 256
1809
+ ]
1810
+ },
1811
+ "unet.up_modules.1.0.blocks.1.block.0.bias": {
1812
+ "dtype": "F32",
1813
+ "shape": [
1814
+ 256
1815
+ ]
1816
+ },
1817
+ "unet.up_modules.1.0.blocks.1.block.0.weight": {
1818
+ "dtype": "F32",
1819
+ "shape": [
1820
+ 256,
1821
+ 256,
1822
+ 5
1823
+ ]
1824
+ },
1825
+ "unet.up_modules.1.0.blocks.1.block.1.bias": {
1826
+ "dtype": "F32",
1827
+ "shape": [
1828
+ 256
1829
+ ]
1830
+ },
1831
+ "unet.up_modules.1.0.blocks.1.block.1.weight": {
1832
+ "dtype": "F32",
1833
+ "shape": [
1834
+ 256
1835
+ ]
1836
+ },
1837
+ "unet.up_modules.1.0.cond_encoder.1.bias": {
1838
+ "dtype": "F32",
1839
+ "shape": [
1840
+ 512
1841
+ ]
1842
+ },
1843
+ "unet.up_modules.1.0.cond_encoder.1.weight": {
1844
+ "dtype": "F32",
1845
+ "shape": [
1846
+ 512,
1847
+ 27776
1848
+ ]
1849
+ },
1850
+ "unet.up_modules.1.0.residual_conv.bias": {
1851
+ "dtype": "F32",
1852
+ "shape": [
1853
+ 256
1854
+ ]
1855
+ },
1856
+ "unet.up_modules.1.0.residual_conv.weight": {
1857
+ "dtype": "F32",
1858
+ "shape": [
1859
+ 256,
1860
+ 1024,
1861
+ 1
1862
+ ]
1863
+ },
1864
+ "unet.up_modules.1.1.blocks.0.block.0.bias": {
1865
+ "dtype": "F32",
1866
+ "shape": [
1867
+ 256
1868
+ ]
1869
+ },
1870
+ "unet.up_modules.1.1.blocks.0.block.0.weight": {
1871
+ "dtype": "F32",
1872
+ "shape": [
1873
+ 256,
1874
+ 256,
1875
+ 5
1876
+ ]
1877
+ },
1878
+ "unet.up_modules.1.1.blocks.0.block.1.bias": {
1879
+ "dtype": "F32",
1880
+ "shape": [
1881
+ 256
1882
+ ]
1883
+ },
1884
+ "unet.up_modules.1.1.blocks.0.block.1.weight": {
1885
+ "dtype": "F32",
1886
+ "shape": [
1887
+ 256
1888
+ ]
1889
+ },
1890
+ "unet.up_modules.1.1.blocks.1.block.0.bias": {
1891
+ "dtype": "F32",
1892
+ "shape": [
1893
+ 256
1894
+ ]
1895
+ },
1896
+ "unet.up_modules.1.1.blocks.1.block.0.weight": {
1897
+ "dtype": "F32",
1898
+ "shape": [
1899
+ 256,
1900
+ 256,
1901
+ 5
1902
+ ]
1903
+ },
1904
+ "unet.up_modules.1.1.blocks.1.block.1.bias": {
1905
+ "dtype": "F32",
1906
+ "shape": [
1907
+ 256
1908
+ ]
1909
+ },
1910
+ "unet.up_modules.1.1.blocks.1.block.1.weight": {
1911
+ "dtype": "F32",
1912
+ "shape": [
1913
+ 256
1914
+ ]
1915
+ },
1916
+ "unet.up_modules.1.1.cond_encoder.1.bias": {
1917
+ "dtype": "F32",
1918
+ "shape": [
1919
+ 512
1920
+ ]
1921
+ },
1922
+ "unet.up_modules.1.1.cond_encoder.1.weight": {
1923
+ "dtype": "F32",
1924
+ "shape": [
1925
+ 512,
1926
+ 27776
1927
+ ]
1928
+ },
1929
+ "unet.up_modules.1.2.bias": {
1930
+ "dtype": "F32",
1931
+ "shape": [
1932
+ 256
1933
+ ]
1934
+ },
1935
+ "unet.up_modules.1.2.weight": {
1936
+ "dtype": "F32",
1937
+ "shape": [
1938
+ 256,
1939
+ 256,
1940
+ 4
1941
+ ]
1942
+ },
1943
+ "vision.head_rgb.blocks.0.attn.proj.bias": {
1944
+ "dtype": "F32",
1945
+ "shape": [
1946
+ 768
1947
+ ]
1948
+ },
1949
+ "vision.head_rgb.blocks.0.attn.proj.weight": {
1950
+ "dtype": "F32",
1951
+ "shape": [
1952
+ 768,
1953
+ 768
1954
+ ]
1955
+ },
1956
+ "vision.head_rgb.blocks.0.attn.qkv.bias": {
1957
+ "dtype": "F32",
1958
+ "shape": [
1959
+ 2304
1960
+ ]
1961
+ },
1962
+ "vision.head_rgb.blocks.0.attn.qkv.weight": {
1963
+ "dtype": "F32",
1964
+ "shape": [
1965
+ 2304,
1966
+ 768
1967
+ ]
1968
+ },
1969
+ "vision.head_rgb.blocks.0.mlp.fc1.bias": {
1970
+ "dtype": "F32",
1971
+ "shape": [
1972
+ 3072
1973
+ ]
1974
+ },
1975
+ "vision.head_rgb.blocks.0.mlp.fc1.weight": {
1976
+ "dtype": "F32",
1977
+ "shape": [
1978
+ 3072,
1979
+ 768
1980
+ ]
1981
+ },
1982
+ "vision.head_rgb.blocks.0.mlp.fc2.bias": {
1983
+ "dtype": "F32",
1984
+ "shape": [
1985
+ 768
1986
+ ]
1987
+ },
1988
+ "vision.head_rgb.blocks.0.mlp.fc2.weight": {
1989
+ "dtype": "F32",
1990
+ "shape": [
1991
+ 768,
1992
+ 3072
1993
+ ]
1994
+ },
1995
+ "vision.head_rgb.blocks.0.norm1.bias": {
1996
+ "dtype": "F32",
1997
+ "shape": [
1998
+ 768
1999
+ ]
2000
+ },
2001
+ "vision.head_rgb.blocks.0.norm1.weight": {
2002
+ "dtype": "F32",
2003
+ "shape": [
2004
+ 768
2005
+ ]
2006
+ },
2007
+ "vision.head_rgb.blocks.0.norm2.bias": {
2008
+ "dtype": "F32",
2009
+ "shape": [
2010
+ 768
2011
+ ]
2012
+ },
2013
+ "vision.head_rgb.blocks.0.norm2.weight": {
2014
+ "dtype": "F32",
2015
+ "shape": [
2016
+ 768
2017
+ ]
2018
+ },
2019
+ "vision.head_rgb.blocks.1.attn.proj.bias": {
2020
+ "dtype": "F32",
2021
+ "shape": [
2022
+ 768
2023
+ ]
2024
+ },
2025
+ "vision.head_rgb.blocks.1.attn.proj.weight": {
2026
+ "dtype": "F32",
2027
+ "shape": [
2028
+ 768,
2029
+ 768
2030
+ ]
2031
+ },
2032
+ "vision.head_rgb.blocks.1.attn.qkv.bias": {
2033
+ "dtype": "F32",
2034
+ "shape": [
2035
+ 2304
2036
+ ]
2037
+ },
2038
+ "vision.head_rgb.blocks.1.attn.qkv.weight": {
2039
+ "dtype": "F32",
2040
+ "shape": [
2041
+ 2304,
2042
+ 768
2043
+ ]
2044
+ },
2045
+ "vision.head_rgb.blocks.1.mlp.fc1.bias": {
2046
+ "dtype": "F32",
2047
+ "shape": [
2048
+ 3072
2049
+ ]
2050
+ },
2051
+ "vision.head_rgb.blocks.1.mlp.fc1.weight": {
2052
+ "dtype": "F32",
2053
+ "shape": [
2054
+ 3072,
2055
+ 768
2056
+ ]
2057
+ },
2058
+ "vision.head_rgb.blocks.1.mlp.fc2.bias": {
2059
+ "dtype": "F32",
2060
+ "shape": [
2061
+ 768
2062
+ ]
2063
+ },
2064
+ "vision.head_rgb.blocks.1.mlp.fc2.weight": {
2065
+ "dtype": "F32",
2066
+ "shape": [
2067
+ 768,
2068
+ 3072
2069
+ ]
2070
+ },
2071
+ "vision.head_rgb.blocks.1.norm1.bias": {
2072
+ "dtype": "F32",
2073
+ "shape": [
2074
+ 768
2075
+ ]
2076
+ },
2077
+ "vision.head_rgb.blocks.1.norm1.weight": {
2078
+ "dtype": "F32",
2079
+ "shape": [
2080
+ 768
2081
+ ]
2082
+ },
2083
+ "vision.head_rgb.blocks.1.norm2.bias": {
2084
+ "dtype": "F32",
2085
+ "shape": [
2086
+ 768
2087
+ ]
2088
+ },
2089
+ "vision.head_rgb.blocks.1.norm2.weight": {
2090
+ "dtype": "F32",
2091
+ "shape": [
2092
+ 768
2093
+ ]
2094
+ },
2095
+ "vision.head_rgb.blocks.10.attn.proj.bias": {
2096
+ "dtype": "F32",
2097
+ "shape": [
2098
+ 768
2099
+ ]
2100
+ },
2101
+ "vision.head_rgb.blocks.10.attn.proj.weight": {
2102
+ "dtype": "F32",
2103
+ "shape": [
2104
+ 768,
2105
+ 768
2106
+ ]
2107
+ },
2108
+ "vision.head_rgb.blocks.10.attn.qkv.bias": {
2109
+ "dtype": "F32",
2110
+ "shape": [
2111
+ 2304
2112
+ ]
2113
+ },
2114
+ "vision.head_rgb.blocks.10.attn.qkv.weight": {
2115
+ "dtype": "F32",
2116
+ "shape": [
2117
+ 2304,
2118
+ 768
2119
+ ]
2120
+ },
2121
+ "vision.head_rgb.blocks.10.mlp.fc1.bias": {
2122
+ "dtype": "F32",
2123
+ "shape": [
2124
+ 3072
2125
+ ]
2126
+ },
2127
+ "vision.head_rgb.blocks.10.mlp.fc1.weight": {
2128
+ "dtype": "F32",
2129
+ "shape": [
2130
+ 3072,
2131
+ 768
2132
+ ]
2133
+ },
2134
+ "vision.head_rgb.blocks.10.mlp.fc2.bias": {
2135
+ "dtype": "F32",
2136
+ "shape": [
2137
+ 768
2138
+ ]
2139
+ },
2140
+ "vision.head_rgb.blocks.10.mlp.fc2.weight": {
2141
+ "dtype": "F32",
2142
+ "shape": [
2143
+ 768,
2144
+ 3072
2145
+ ]
2146
+ },
2147
+ "vision.head_rgb.blocks.10.norm1.bias": {
2148
+ "dtype": "F32",
2149
+ "shape": [
2150
+ 768
2151
+ ]
2152
+ },
2153
+ "vision.head_rgb.blocks.10.norm1.weight": {
2154
+ "dtype": "F32",
2155
+ "shape": [
2156
+ 768
2157
+ ]
2158
+ },
2159
+ "vision.head_rgb.blocks.10.norm2.bias": {
2160
+ "dtype": "F32",
2161
+ "shape": [
2162
+ 768
2163
+ ]
2164
+ },
2165
+ "vision.head_rgb.blocks.10.norm2.weight": {
2166
+ "dtype": "F32",
2167
+ "shape": [
2168
+ 768
2169
+ ]
2170
+ },
2171
+ "vision.head_rgb.blocks.11.attn.proj.bias": {
2172
+ "dtype": "F32",
2173
+ "shape": [
2174
+ 768
2175
+ ]
2176
+ },
2177
+ "vision.head_rgb.blocks.11.attn.proj.weight": {
2178
+ "dtype": "F32",
2179
+ "shape": [
2180
+ 768,
2181
+ 768
2182
+ ]
2183
+ },
2184
+ "vision.head_rgb.blocks.11.attn.qkv.bias": {
2185
+ "dtype": "F32",
2186
+ "shape": [
2187
+ 2304
2188
+ ]
2189
+ },
2190
+ "vision.head_rgb.blocks.11.attn.qkv.weight": {
2191
+ "dtype": "F32",
2192
+ "shape": [
2193
+ 2304,
2194
+ 768
2195
+ ]
2196
+ },
2197
+ "vision.head_rgb.blocks.11.mlp.fc1.bias": {
2198
+ "dtype": "F32",
2199
+ "shape": [
2200
+ 3072
2201
+ ]
2202
+ },
2203
+ "vision.head_rgb.blocks.11.mlp.fc1.weight": {
2204
+ "dtype": "F32",
2205
+ "shape": [
2206
+ 3072,
2207
+ 768
2208
+ ]
2209
+ },
2210
+ "vision.head_rgb.blocks.11.mlp.fc2.bias": {
2211
+ "dtype": "F32",
2212
+ "shape": [
2213
+ 768
2214
+ ]
2215
+ },
2216
+ "vision.head_rgb.blocks.11.mlp.fc2.weight": {
2217
+ "dtype": "F32",
2218
+ "shape": [
2219
+ 768,
2220
+ 3072
2221
+ ]
2222
+ },
2223
+ "vision.head_rgb.blocks.11.norm1.bias": {
2224
+ "dtype": "F32",
2225
+ "shape": [
2226
+ 768
2227
+ ]
2228
+ },
2229
+ "vision.head_rgb.blocks.11.norm1.weight": {
2230
+ "dtype": "F32",
2231
+ "shape": [
2232
+ 768
2233
+ ]
2234
+ },
2235
+ "vision.head_rgb.blocks.11.norm2.bias": {
2236
+ "dtype": "F32",
2237
+ "shape": [
2238
+ 768
2239
+ ]
2240
+ },
2241
+ "vision.head_rgb.blocks.11.norm2.weight": {
2242
+ "dtype": "F32",
2243
+ "shape": [
2244
+ 768
2245
+ ]
2246
+ },
2247
+ "vision.head_rgb.blocks.2.attn.proj.bias": {
2248
+ "dtype": "F32",
2249
+ "shape": [
2250
+ 768
2251
+ ]
2252
+ },
2253
+ "vision.head_rgb.blocks.2.attn.proj.weight": {
2254
+ "dtype": "F32",
2255
+ "shape": [
2256
+ 768,
2257
+ 768
2258
+ ]
2259
+ },
2260
+ "vision.head_rgb.blocks.2.attn.qkv.bias": {
2261
+ "dtype": "F32",
2262
+ "shape": [
2263
+ 2304
2264
+ ]
2265
+ },
2266
+ "vision.head_rgb.blocks.2.attn.qkv.weight": {
2267
+ "dtype": "F32",
2268
+ "shape": [
2269
+ 2304,
2270
+ 768
2271
+ ]
2272
+ },
2273
+ "vision.head_rgb.blocks.2.mlp.fc1.bias": {
2274
+ "dtype": "F32",
2275
+ "shape": [
2276
+ 3072
2277
+ ]
2278
+ },
2279
+ "vision.head_rgb.blocks.2.mlp.fc1.weight": {
2280
+ "dtype": "F32",
2281
+ "shape": [
2282
+ 3072,
2283
+ 768
2284
+ ]
2285
+ },
2286
+ "vision.head_rgb.blocks.2.mlp.fc2.bias": {
2287
+ "dtype": "F32",
2288
+ "shape": [
2289
+ 768
2290
+ ]
2291
+ },
2292
+ "vision.head_rgb.blocks.2.mlp.fc2.weight": {
2293
+ "dtype": "F32",
2294
+ "shape": [
2295
+ 768,
2296
+ 3072
2297
+ ]
2298
+ },
2299
+ "vision.head_rgb.blocks.2.norm1.bias": {
2300
+ "dtype": "F32",
2301
+ "shape": [
2302
+ 768
2303
+ ]
2304
+ },
2305
+ "vision.head_rgb.blocks.2.norm1.weight": {
2306
+ "dtype": "F32",
2307
+ "shape": [
2308
+ 768
2309
+ ]
2310
+ },
2311
+ "vision.head_rgb.blocks.2.norm2.bias": {
2312
+ "dtype": "F32",
2313
+ "shape": [
2314
+ 768
2315
+ ]
2316
+ },
2317
+ "vision.head_rgb.blocks.2.norm2.weight": {
2318
+ "dtype": "F32",
2319
+ "shape": [
2320
+ 768
2321
+ ]
2322
+ },
2323
+ "vision.head_rgb.blocks.3.attn.proj.bias": {
2324
+ "dtype": "F32",
2325
+ "shape": [
2326
+ 768
2327
+ ]
2328
+ },
2329
+ "vision.head_rgb.blocks.3.attn.proj.weight": {
2330
+ "dtype": "F32",
2331
+ "shape": [
2332
+ 768,
2333
+ 768
2334
+ ]
2335
+ },
2336
+ "vision.head_rgb.blocks.3.attn.qkv.bias": {
2337
+ "dtype": "F32",
2338
+ "shape": [
2339
+ 2304
2340
+ ]
2341
+ },
2342
+ "vision.head_rgb.blocks.3.attn.qkv.weight": {
2343
+ "dtype": "F32",
2344
+ "shape": [
2345
+ 2304,
2346
+ 768
2347
+ ]
2348
+ },
2349
+ "vision.head_rgb.blocks.3.mlp.fc1.bias": {
2350
+ "dtype": "F32",
2351
+ "shape": [
2352
+ 3072
2353
+ ]
2354
+ },
2355
+ "vision.head_rgb.blocks.3.mlp.fc1.weight": {
2356
+ "dtype": "F32",
2357
+ "shape": [
2358
+ 3072,
2359
+ 768
2360
+ ]
2361
+ },
2362
+ "vision.head_rgb.blocks.3.mlp.fc2.bias": {
2363
+ "dtype": "F32",
2364
+ "shape": [
2365
+ 768
2366
+ ]
2367
+ },
2368
+ "vision.head_rgb.blocks.3.mlp.fc2.weight": {
2369
+ "dtype": "F32",
2370
+ "shape": [
2371
+ 768,
2372
+ 3072
2373
+ ]
2374
+ },
2375
+ "vision.head_rgb.blocks.3.norm1.bias": {
2376
+ "dtype": "F32",
2377
+ "shape": [
2378
+ 768
2379
+ ]
2380
+ },
2381
+ "vision.head_rgb.blocks.3.norm1.weight": {
2382
+ "dtype": "F32",
2383
+ "shape": [
2384
+ 768
2385
+ ]
2386
+ },
2387
+ "vision.head_rgb.blocks.3.norm2.bias": {
2388
+ "dtype": "F32",
2389
+ "shape": [
2390
+ 768
2391
+ ]
2392
+ },
2393
+ "vision.head_rgb.blocks.3.norm2.weight": {
2394
+ "dtype": "F32",
2395
+ "shape": [
2396
+ 768
2397
+ ]
2398
+ },
2399
+ "vision.head_rgb.blocks.4.attn.proj.bias": {
2400
+ "dtype": "F32",
2401
+ "shape": [
2402
+ 768
2403
+ ]
2404
+ },
2405
+ "vision.head_rgb.blocks.4.attn.proj.weight": {
2406
+ "dtype": "F32",
2407
+ "shape": [
2408
+ 768,
2409
+ 768
2410
+ ]
2411
+ },
2412
+ "vision.head_rgb.blocks.4.attn.qkv.bias": {
2413
+ "dtype": "F32",
2414
+ "shape": [
2415
+ 2304
2416
+ ]
2417
+ },
2418
+ "vision.head_rgb.blocks.4.attn.qkv.weight": {
2419
+ "dtype": "F32",
2420
+ "shape": [
2421
+ 2304,
2422
+ 768
2423
+ ]
2424
+ },
2425
+ "vision.head_rgb.blocks.4.mlp.fc1.bias": {
2426
+ "dtype": "F32",
2427
+ "shape": [
2428
+ 3072
2429
+ ]
2430
+ },
2431
+ "vision.head_rgb.blocks.4.mlp.fc1.weight": {
2432
+ "dtype": "F32",
2433
+ "shape": [
2434
+ 3072,
2435
+ 768
2436
+ ]
2437
+ },
2438
+ "vision.head_rgb.blocks.4.mlp.fc2.bias": {
2439
+ "dtype": "F32",
2440
+ "shape": [
2441
+ 768
2442
+ ]
2443
+ },
2444
+ "vision.head_rgb.blocks.4.mlp.fc2.weight": {
2445
+ "dtype": "F32",
2446
+ "shape": [
2447
+ 768,
2448
+ 3072
2449
+ ]
2450
+ },
2451
+ "vision.head_rgb.blocks.4.norm1.bias": {
2452
+ "dtype": "F32",
2453
+ "shape": [
2454
+ 768
2455
+ ]
2456
+ },
2457
+ "vision.head_rgb.blocks.4.norm1.weight": {
2458
+ "dtype": "F32",
2459
+ "shape": [
2460
+ 768
2461
+ ]
2462
+ },
2463
+ "vision.head_rgb.blocks.4.norm2.bias": {
2464
+ "dtype": "F32",
2465
+ "shape": [
2466
+ 768
2467
+ ]
2468
+ },
2469
+ "vision.head_rgb.blocks.4.norm2.weight": {
2470
+ "dtype": "F32",
2471
+ "shape": [
2472
+ 768
2473
+ ]
2474
+ },
2475
+ "vision.head_rgb.blocks.5.attn.proj.bias": {
2476
+ "dtype": "F32",
2477
+ "shape": [
2478
+ 768
2479
+ ]
2480
+ },
2481
+ "vision.head_rgb.blocks.5.attn.proj.weight": {
2482
+ "dtype": "F32",
2483
+ "shape": [
2484
+ 768,
2485
+ 768
2486
+ ]
2487
+ },
2488
+ "vision.head_rgb.blocks.5.attn.qkv.bias": {
2489
+ "dtype": "F32",
2490
+ "shape": [
2491
+ 2304
2492
+ ]
2493
+ },
2494
+ "vision.head_rgb.blocks.5.attn.qkv.weight": {
2495
+ "dtype": "F32",
2496
+ "shape": [
2497
+ 2304,
2498
+ 768
2499
+ ]
2500
+ },
2501
+ "vision.head_rgb.blocks.5.mlp.fc1.bias": {
2502
+ "dtype": "F32",
2503
+ "shape": [
2504
+ 3072
2505
+ ]
2506
+ },
2507
+ "vision.head_rgb.blocks.5.mlp.fc1.weight": {
2508
+ "dtype": "F32",
2509
+ "shape": [
2510
+ 3072,
2511
+ 768
2512
+ ]
2513
+ },
2514
+ "vision.head_rgb.blocks.5.mlp.fc2.bias": {
2515
+ "dtype": "F32",
2516
+ "shape": [
2517
+ 768
2518
+ ]
2519
+ },
2520
+ "vision.head_rgb.blocks.5.mlp.fc2.weight": {
2521
+ "dtype": "F32",
2522
+ "shape": [
2523
+ 768,
2524
+ 3072
2525
+ ]
2526
+ },
2527
+ "vision.head_rgb.blocks.5.norm1.bias": {
2528
+ "dtype": "F32",
2529
+ "shape": [
2530
+ 768
2531
+ ]
2532
+ },
2533
+ "vision.head_rgb.blocks.5.norm1.weight": {
2534
+ "dtype": "F32",
2535
+ "shape": [
2536
+ 768
2537
+ ]
2538
+ },
2539
+ "vision.head_rgb.blocks.5.norm2.bias": {
2540
+ "dtype": "F32",
2541
+ "shape": [
2542
+ 768
2543
+ ]
2544
+ },
2545
+ "vision.head_rgb.blocks.5.norm2.weight": {
2546
+ "dtype": "F32",
2547
+ "shape": [
2548
+ 768
2549
+ ]
2550
+ },
2551
+ "vision.head_rgb.blocks.6.attn.proj.bias": {
2552
+ "dtype": "F32",
2553
+ "shape": [
2554
+ 768
2555
+ ]
2556
+ },
2557
+ "vision.head_rgb.blocks.6.attn.proj.weight": {
2558
+ "dtype": "F32",
2559
+ "shape": [
2560
+ 768,
2561
+ 768
2562
+ ]
2563
+ },
2564
+ "vision.head_rgb.blocks.6.attn.qkv.bias": {
2565
+ "dtype": "F32",
2566
+ "shape": [
2567
+ 2304
2568
+ ]
2569
+ },
2570
+ "vision.head_rgb.blocks.6.attn.qkv.weight": {
2571
+ "dtype": "F32",
2572
+ "shape": [
2573
+ 2304,
2574
+ 768
2575
+ ]
2576
+ },
2577
+ "vision.head_rgb.blocks.6.mlp.fc1.bias": {
2578
+ "dtype": "F32",
2579
+ "shape": [
2580
+ 3072
2581
+ ]
2582
+ },
2583
+ "vision.head_rgb.blocks.6.mlp.fc1.weight": {
2584
+ "dtype": "F32",
2585
+ "shape": [
2586
+ 3072,
2587
+ 768
2588
+ ]
2589
+ },
2590
+ "vision.head_rgb.blocks.6.mlp.fc2.bias": {
2591
+ "dtype": "F32",
2592
+ "shape": [
2593
+ 768
2594
+ ]
2595
+ },
2596
+ "vision.head_rgb.blocks.6.mlp.fc2.weight": {
2597
+ "dtype": "F32",
2598
+ "shape": [
2599
+ 768,
2600
+ 3072
2601
+ ]
2602
+ },
2603
+ "vision.head_rgb.blocks.6.norm1.bias": {
2604
+ "dtype": "F32",
2605
+ "shape": [
2606
+ 768
2607
+ ]
2608
+ },
2609
+ "vision.head_rgb.blocks.6.norm1.weight": {
2610
+ "dtype": "F32",
2611
+ "shape": [
2612
+ 768
2613
+ ]
2614
+ },
2615
+ "vision.head_rgb.blocks.6.norm2.bias": {
2616
+ "dtype": "F32",
2617
+ "shape": [
2618
+ 768
2619
+ ]
2620
+ },
2621
+ "vision.head_rgb.blocks.6.norm2.weight": {
2622
+ "dtype": "F32",
2623
+ "shape": [
2624
+ 768
2625
+ ]
2626
+ },
2627
+ "vision.head_rgb.blocks.7.attn.proj.bias": {
2628
+ "dtype": "F32",
2629
+ "shape": [
2630
+ 768
2631
+ ]
2632
+ },
2633
+ "vision.head_rgb.blocks.7.attn.proj.weight": {
2634
+ "dtype": "F32",
2635
+ "shape": [
2636
+ 768,
2637
+ 768
2638
+ ]
2639
+ },
2640
+ "vision.head_rgb.blocks.7.attn.qkv.bias": {
2641
+ "dtype": "F32",
2642
+ "shape": [
2643
+ 2304
2644
+ ]
2645
+ },
2646
+ "vision.head_rgb.blocks.7.attn.qkv.weight": {
2647
+ "dtype": "F32",
2648
+ "shape": [
2649
+ 2304,
2650
+ 768
2651
+ ]
2652
+ },
2653
+ "vision.head_rgb.blocks.7.mlp.fc1.bias": {
2654
+ "dtype": "F32",
2655
+ "shape": [
2656
+ 3072
2657
+ ]
2658
+ },
2659
+ "vision.head_rgb.blocks.7.mlp.fc1.weight": {
2660
+ "dtype": "F32",
2661
+ "shape": [
2662
+ 3072,
2663
+ 768
2664
+ ]
2665
+ },
2666
+ "vision.head_rgb.blocks.7.mlp.fc2.bias": {
2667
+ "dtype": "F32",
2668
+ "shape": [
2669
+ 768
2670
+ ]
2671
+ },
2672
+ "vision.head_rgb.blocks.7.mlp.fc2.weight": {
2673
+ "dtype": "F32",
2674
+ "shape": [
2675
+ 768,
2676
+ 3072
2677
+ ]
2678
+ },
2679
+ "vision.head_rgb.blocks.7.norm1.bias": {
2680
+ "dtype": "F32",
2681
+ "shape": [
2682
+ 768
2683
+ ]
2684
+ },
2685
+ "vision.head_rgb.blocks.7.norm1.weight": {
2686
+ "dtype": "F32",
2687
+ "shape": [
2688
+ 768
2689
+ ]
2690
+ },
2691
+ "vision.head_rgb.blocks.7.norm2.bias": {
2692
+ "dtype": "F32",
2693
+ "shape": [
2694
+ 768
2695
+ ]
2696
+ },
2697
+ "vision.head_rgb.blocks.7.norm2.weight": {
2698
+ "dtype": "F32",
2699
+ "shape": [
2700
+ 768
2701
+ ]
2702
+ },
2703
+ "vision.head_rgb.blocks.8.attn.proj.bias": {
2704
+ "dtype": "F32",
2705
+ "shape": [
2706
+ 768
2707
+ ]
2708
+ },
2709
+ "vision.head_rgb.blocks.8.attn.proj.weight": {
2710
+ "dtype": "F32",
2711
+ "shape": [
2712
+ 768,
2713
+ 768
2714
+ ]
2715
+ },
2716
+ "vision.head_rgb.blocks.8.attn.qkv.bias": {
2717
+ "dtype": "F32",
2718
+ "shape": [
2719
+ 2304
2720
+ ]
2721
+ },
2722
+ "vision.head_rgb.blocks.8.attn.qkv.weight": {
2723
+ "dtype": "F32",
2724
+ "shape": [
2725
+ 2304,
2726
+ 768
2727
+ ]
2728
+ },
2729
+ "vision.head_rgb.blocks.8.mlp.fc1.bias": {
2730
+ "dtype": "F32",
2731
+ "shape": [
2732
+ 3072
2733
+ ]
2734
+ },
2735
+ "vision.head_rgb.blocks.8.mlp.fc1.weight": {
2736
+ "dtype": "F32",
2737
+ "shape": [
2738
+ 3072,
2739
+ 768
2740
+ ]
2741
+ },
2742
+ "vision.head_rgb.blocks.8.mlp.fc2.bias": {
2743
+ "dtype": "F32",
2744
+ "shape": [
2745
+ 768
2746
+ ]
2747
+ },
2748
+ "vision.head_rgb.blocks.8.mlp.fc2.weight": {
2749
+ "dtype": "F32",
2750
+ "shape": [
2751
+ 768,
2752
+ 3072
2753
+ ]
2754
+ },
2755
+ "vision.head_rgb.blocks.8.norm1.bias": {
2756
+ "dtype": "F32",
2757
+ "shape": [
2758
+ 768
2759
+ ]
2760
+ },
2761
+ "vision.head_rgb.blocks.8.norm1.weight": {
2762
+ "dtype": "F32",
2763
+ "shape": [
2764
+ 768
2765
+ ]
2766
+ },
2767
+ "vision.head_rgb.blocks.8.norm2.bias": {
2768
+ "dtype": "F32",
2769
+ "shape": [
2770
+ 768
2771
+ ]
2772
+ },
2773
+ "vision.head_rgb.blocks.8.norm2.weight": {
2774
+ "dtype": "F32",
2775
+ "shape": [
2776
+ 768
2777
+ ]
2778
+ },
2779
+ "vision.head_rgb.blocks.9.attn.proj.bias": {
2780
+ "dtype": "F32",
2781
+ "shape": [
2782
+ 768
2783
+ ]
2784
+ },
2785
+ "vision.head_rgb.blocks.9.attn.proj.weight": {
2786
+ "dtype": "F32",
2787
+ "shape": [
2788
+ 768,
2789
+ 768
2790
+ ]
2791
+ },
2792
+ "vision.head_rgb.blocks.9.attn.qkv.bias": {
2793
+ "dtype": "F32",
2794
+ "shape": [
2795
+ 2304
2796
+ ]
2797
+ },
2798
+ "vision.head_rgb.blocks.9.attn.qkv.weight": {
2799
+ "dtype": "F32",
2800
+ "shape": [
2801
+ 2304,
2802
+ 768
2803
+ ]
2804
+ },
2805
+ "vision.head_rgb.blocks.9.mlp.fc1.bias": {
2806
+ "dtype": "F32",
2807
+ "shape": [
2808
+ 3072
2809
+ ]
2810
+ },
2811
+ "vision.head_rgb.blocks.9.mlp.fc1.weight": {
2812
+ "dtype": "F32",
2813
+ "shape": [
2814
+ 3072,
2815
+ 768
2816
+ ]
2817
+ },
2818
+ "vision.head_rgb.blocks.9.mlp.fc2.bias": {
2819
+ "dtype": "F32",
2820
+ "shape": [
2821
+ 768
2822
+ ]
2823
+ },
2824
+ "vision.head_rgb.blocks.9.mlp.fc2.weight": {
2825
+ "dtype": "F32",
2826
+ "shape": [
2827
+ 768,
2828
+ 3072
2829
+ ]
2830
+ },
2831
+ "vision.head_rgb.blocks.9.norm1.bias": {
2832
+ "dtype": "F32",
2833
+ "shape": [
2834
+ 768
2835
+ ]
2836
+ },
2837
+ "vision.head_rgb.blocks.9.norm1.weight": {
2838
+ "dtype": "F32",
2839
+ "shape": [
2840
+ 768
2841
+ ]
2842
+ },
2843
+ "vision.head_rgb.blocks.9.norm2.bias": {
2844
+ "dtype": "F32",
2845
+ "shape": [
2846
+ 768
2847
+ ]
2848
+ },
2849
+ "vision.head_rgb.blocks.9.norm2.weight": {
2850
+ "dtype": "F32",
2851
+ "shape": [
2852
+ 768
2853
+ ]
2854
+ },
2855
+ "vision.head_rgb.cls_token": {
2856
+ "dtype": "F32",
2857
+ "shape": [
2858
+ 1,
2859
+ 1,
2860
+ 768
2861
+ ]
2862
+ },
2863
+ "vision.head_rgb.norm.bias": {
2864
+ "dtype": "F32",
2865
+ "shape": [
2866
+ 768
2867
+ ]
2868
+ },
2869
+ "vision.head_rgb.norm.weight": {
2870
+ "dtype": "F32",
2871
+ "shape": [
2872
+ 768
2873
+ ]
2874
+ },
2875
+ "vision.head_rgb.norm_pre.bias": {
2876
+ "dtype": "F32",
2877
+ "shape": [
2878
+ 768
2879
+ ]
2880
+ },
2881
+ "vision.head_rgb.norm_pre.weight": {
2882
+ "dtype": "F32",
2883
+ "shape": [
2884
+ 768
2885
+ ]
2886
+ },
2887
+ "vision.head_rgb.patch_embed.proj.weight": {
2888
+ "dtype": "F32",
2889
+ "shape": [
2890
+ 768,
2891
+ 3,
2892
+ 16,
2893
+ 16
2894
+ ]
2895
+ },
2896
+ "vision.head_rgb.pos_embed": {
2897
+ "dtype": "F32",
2898
+ "shape": [
2899
+ 1,
2900
+ 197,
2901
+ 768
2902
+ ]
2903
+ },
2904
+ "vision.left_wrist_rgb.blocks.0.attn.proj.bias": {
2905
+ "dtype": "F32",
2906
+ "shape": [
2907
+ 768
2908
+ ]
2909
+ },
2910
+ "vision.left_wrist_rgb.blocks.0.attn.proj.weight": {
2911
+ "dtype": "F32",
2912
+ "shape": [
2913
+ 768,
2914
+ 768
2915
+ ]
2916
+ },
2917
+ "vision.left_wrist_rgb.blocks.0.attn.qkv.bias": {
2918
+ "dtype": "F32",
2919
+ "shape": [
2920
+ 2304
2921
+ ]
2922
+ },
2923
+ "vision.left_wrist_rgb.blocks.0.attn.qkv.weight": {
2924
+ "dtype": "F32",
2925
+ "shape": [
2926
+ 2304,
2927
+ 768
2928
+ ]
2929
+ },
2930
+ "vision.left_wrist_rgb.blocks.0.mlp.fc1.bias": {
2931
+ "dtype": "F32",
2932
+ "shape": [
2933
+ 3072
2934
+ ]
2935
+ },
2936
+ "vision.left_wrist_rgb.blocks.0.mlp.fc1.weight": {
2937
+ "dtype": "F32",
2938
+ "shape": [
2939
+ 3072,
2940
+ 768
2941
+ ]
2942
+ },
2943
+ "vision.left_wrist_rgb.blocks.0.mlp.fc2.bias": {
2944
+ "dtype": "F32",
2945
+ "shape": [
2946
+ 768
2947
+ ]
2948
+ },
2949
+ "vision.left_wrist_rgb.blocks.0.mlp.fc2.weight": {
2950
+ "dtype": "F32",
2951
+ "shape": [
2952
+ 768,
2953
+ 3072
2954
+ ]
2955
+ },
2956
+ "vision.left_wrist_rgb.blocks.0.norm1.bias": {
2957
+ "dtype": "F32",
2958
+ "shape": [
2959
+ 768
2960
+ ]
2961
+ },
2962
+ "vision.left_wrist_rgb.blocks.0.norm1.weight": {
2963
+ "dtype": "F32",
2964
+ "shape": [
2965
+ 768
2966
+ ]
2967
+ },
2968
+ "vision.left_wrist_rgb.blocks.0.norm2.bias": {
2969
+ "dtype": "F32",
2970
+ "shape": [
2971
+ 768
2972
+ ]
2973
+ },
2974
+ "vision.left_wrist_rgb.blocks.0.norm2.weight": {
2975
+ "dtype": "F32",
2976
+ "shape": [
2977
+ 768
2978
+ ]
2979
+ },
2980
+ "vision.left_wrist_rgb.blocks.1.attn.proj.bias": {
2981
+ "dtype": "F32",
2982
+ "shape": [
2983
+ 768
2984
+ ]
2985
+ },
2986
+ "vision.left_wrist_rgb.blocks.1.attn.proj.weight": {
2987
+ "dtype": "F32",
2988
+ "shape": [
2989
+ 768,
2990
+ 768
2991
+ ]
2992
+ },
2993
+ "vision.left_wrist_rgb.blocks.1.attn.qkv.bias": {
2994
+ "dtype": "F32",
2995
+ "shape": [
2996
+ 2304
2997
+ ]
2998
+ },
2999
+ "vision.left_wrist_rgb.blocks.1.attn.qkv.weight": {
3000
+ "dtype": "F32",
3001
+ "shape": [
3002
+ 2304,
3003
+ 768
3004
+ ]
3005
+ },
3006
+ "vision.left_wrist_rgb.blocks.1.mlp.fc1.bias": {
3007
+ "dtype": "F32",
3008
+ "shape": [
3009
+ 3072
3010
+ ]
3011
+ },
3012
+ "vision.left_wrist_rgb.blocks.1.mlp.fc1.weight": {
3013
+ "dtype": "F32",
3014
+ "shape": [
3015
+ 3072,
3016
+ 768
3017
+ ]
3018
+ },
3019
+ "vision.left_wrist_rgb.blocks.1.mlp.fc2.bias": {
3020
+ "dtype": "F32",
3021
+ "shape": [
3022
+ 768
3023
+ ]
3024
+ },
3025
+ "vision.left_wrist_rgb.blocks.1.mlp.fc2.weight": {
3026
+ "dtype": "F32",
3027
+ "shape": [
3028
+ 768,
3029
+ 3072
3030
+ ]
3031
+ },
3032
+ "vision.left_wrist_rgb.blocks.1.norm1.bias": {
3033
+ "dtype": "F32",
3034
+ "shape": [
3035
+ 768
3036
+ ]
3037
+ },
3038
+ "vision.left_wrist_rgb.blocks.1.norm1.weight": {
3039
+ "dtype": "F32",
3040
+ "shape": [
3041
+ 768
3042
+ ]
3043
+ },
3044
+ "vision.left_wrist_rgb.blocks.1.norm2.bias": {
3045
+ "dtype": "F32",
3046
+ "shape": [
3047
+ 768
3048
+ ]
3049
+ },
3050
+ "vision.left_wrist_rgb.blocks.1.norm2.weight": {
3051
+ "dtype": "F32",
3052
+ "shape": [
3053
+ 768
3054
+ ]
3055
+ },
3056
+ "vision.left_wrist_rgb.blocks.10.attn.proj.bias": {
3057
+ "dtype": "F32",
3058
+ "shape": [
3059
+ 768
3060
+ ]
3061
+ },
3062
+ "vision.left_wrist_rgb.blocks.10.attn.proj.weight": {
3063
+ "dtype": "F32",
3064
+ "shape": [
3065
+ 768,
3066
+ 768
3067
+ ]
3068
+ },
3069
+ "vision.left_wrist_rgb.blocks.10.attn.qkv.bias": {
3070
+ "dtype": "F32",
3071
+ "shape": [
3072
+ 2304
3073
+ ]
3074
+ },
3075
+ "vision.left_wrist_rgb.blocks.10.attn.qkv.weight": {
3076
+ "dtype": "F32",
3077
+ "shape": [
3078
+ 2304,
3079
+ 768
3080
+ ]
3081
+ },
3082
+ "vision.left_wrist_rgb.blocks.10.mlp.fc1.bias": {
3083
+ "dtype": "F32",
3084
+ "shape": [
3085
+ 3072
3086
+ ]
3087
+ },
3088
+ "vision.left_wrist_rgb.blocks.10.mlp.fc1.weight": {
3089
+ "dtype": "F32",
3090
+ "shape": [
3091
+ 3072,
3092
+ 768
3093
+ ]
3094
+ },
3095
+ "vision.left_wrist_rgb.blocks.10.mlp.fc2.bias": {
3096
+ "dtype": "F32",
3097
+ "shape": [
3098
+ 768
3099
+ ]
3100
+ },
3101
+ "vision.left_wrist_rgb.blocks.10.mlp.fc2.weight": {
3102
+ "dtype": "F32",
3103
+ "shape": [
3104
+ 768,
3105
+ 3072
3106
+ ]
3107
+ },
3108
+ "vision.left_wrist_rgb.blocks.10.norm1.bias": {
3109
+ "dtype": "F32",
3110
+ "shape": [
3111
+ 768
3112
+ ]
3113
+ },
3114
+ "vision.left_wrist_rgb.blocks.10.norm1.weight": {
3115
+ "dtype": "F32",
3116
+ "shape": [
3117
+ 768
3118
+ ]
3119
+ },
3120
+ "vision.left_wrist_rgb.blocks.10.norm2.bias": {
3121
+ "dtype": "F32",
3122
+ "shape": [
3123
+ 768
3124
+ ]
3125
+ },
3126
+ "vision.left_wrist_rgb.blocks.10.norm2.weight": {
3127
+ "dtype": "F32",
3128
+ "shape": [
3129
+ 768
3130
+ ]
3131
+ },
3132
+ "vision.left_wrist_rgb.blocks.11.attn.proj.bias": {
3133
+ "dtype": "F32",
3134
+ "shape": [
3135
+ 768
3136
+ ]
3137
+ },
3138
+ "vision.left_wrist_rgb.blocks.11.attn.proj.weight": {
3139
+ "dtype": "F32",
3140
+ "shape": [
3141
+ 768,
3142
+ 768
3143
+ ]
3144
+ },
3145
+ "vision.left_wrist_rgb.blocks.11.attn.qkv.bias": {
3146
+ "dtype": "F32",
3147
+ "shape": [
3148
+ 2304
3149
+ ]
3150
+ },
3151
+ "vision.left_wrist_rgb.blocks.11.attn.qkv.weight": {
3152
+ "dtype": "F32",
3153
+ "shape": [
3154
+ 2304,
3155
+ 768
3156
+ ]
3157
+ },
3158
+ "vision.left_wrist_rgb.blocks.11.mlp.fc1.bias": {
3159
+ "dtype": "F32",
3160
+ "shape": [
3161
+ 3072
3162
+ ]
3163
+ },
3164
+ "vision.left_wrist_rgb.blocks.11.mlp.fc1.weight": {
3165
+ "dtype": "F32",
3166
+ "shape": [
3167
+ 3072,
3168
+ 768
3169
+ ]
3170
+ },
3171
+ "vision.left_wrist_rgb.blocks.11.mlp.fc2.bias": {
3172
+ "dtype": "F32",
3173
+ "shape": [
3174
+ 768
3175
+ ]
3176
+ },
3177
+ "vision.left_wrist_rgb.blocks.11.mlp.fc2.weight": {
3178
+ "dtype": "F32",
3179
+ "shape": [
3180
+ 768,
3181
+ 3072
3182
+ ]
3183
+ },
3184
+ "vision.left_wrist_rgb.blocks.11.norm1.bias": {
3185
+ "dtype": "F32",
3186
+ "shape": [
3187
+ 768
3188
+ ]
3189
+ },
3190
+ "vision.left_wrist_rgb.blocks.11.norm1.weight": {
3191
+ "dtype": "F32",
3192
+ "shape": [
3193
+ 768
3194
+ ]
3195
+ },
3196
+ "vision.left_wrist_rgb.blocks.11.norm2.bias": {
3197
+ "dtype": "F32",
3198
+ "shape": [
3199
+ 768
3200
+ ]
3201
+ },
3202
+ "vision.left_wrist_rgb.blocks.11.norm2.weight": {
3203
+ "dtype": "F32",
3204
+ "shape": [
3205
+ 768
3206
+ ]
3207
+ },
3208
+ "vision.left_wrist_rgb.blocks.2.attn.proj.bias": {
3209
+ "dtype": "F32",
3210
+ "shape": [
3211
+ 768
3212
+ ]
3213
+ },
3214
+ "vision.left_wrist_rgb.blocks.2.attn.proj.weight": {
3215
+ "dtype": "F32",
3216
+ "shape": [
3217
+ 768,
3218
+ 768
3219
+ ]
3220
+ },
3221
+ "vision.left_wrist_rgb.blocks.2.attn.qkv.bias": {
3222
+ "dtype": "F32",
3223
+ "shape": [
3224
+ 2304
3225
+ ]
3226
+ },
3227
+ "vision.left_wrist_rgb.blocks.2.attn.qkv.weight": {
3228
+ "dtype": "F32",
3229
+ "shape": [
3230
+ 2304,
3231
+ 768
3232
+ ]
3233
+ },
3234
+ "vision.left_wrist_rgb.blocks.2.mlp.fc1.bias": {
3235
+ "dtype": "F32",
3236
+ "shape": [
3237
+ 3072
3238
+ ]
3239
+ },
3240
+ "vision.left_wrist_rgb.blocks.2.mlp.fc1.weight": {
3241
+ "dtype": "F32",
3242
+ "shape": [
3243
+ 3072,
3244
+ 768
3245
+ ]
3246
+ },
3247
+ "vision.left_wrist_rgb.blocks.2.mlp.fc2.bias": {
3248
+ "dtype": "F32",
3249
+ "shape": [
3250
+ 768
3251
+ ]
3252
+ },
3253
+ "vision.left_wrist_rgb.blocks.2.mlp.fc2.weight": {
3254
+ "dtype": "F32",
3255
+ "shape": [
3256
+ 768,
3257
+ 3072
3258
+ ]
3259
+ },
3260
+ "vision.left_wrist_rgb.blocks.2.norm1.bias": {
3261
+ "dtype": "F32",
3262
+ "shape": [
3263
+ 768
3264
+ ]
3265
+ },
3266
+ "vision.left_wrist_rgb.blocks.2.norm1.weight": {
3267
+ "dtype": "F32",
3268
+ "shape": [
3269
+ 768
3270
+ ]
3271
+ },
3272
+ "vision.left_wrist_rgb.blocks.2.norm2.bias": {
3273
+ "dtype": "F32",
3274
+ "shape": [
3275
+ 768
3276
+ ]
3277
+ },
3278
+ "vision.left_wrist_rgb.blocks.2.norm2.weight": {
3279
+ "dtype": "F32",
3280
+ "shape": [
3281
+ 768
3282
+ ]
3283
+ },
3284
+ "vision.left_wrist_rgb.blocks.3.attn.proj.bias": {
3285
+ "dtype": "F32",
3286
+ "shape": [
3287
+ 768
3288
+ ]
3289
+ },
3290
+ "vision.left_wrist_rgb.blocks.3.attn.proj.weight": {
3291
+ "dtype": "F32",
3292
+ "shape": [
3293
+ 768,
3294
+ 768
3295
+ ]
3296
+ },
3297
+ "vision.left_wrist_rgb.blocks.3.attn.qkv.bias": {
3298
+ "dtype": "F32",
3299
+ "shape": [
3300
+ 2304
3301
+ ]
3302
+ },
3303
+ "vision.left_wrist_rgb.blocks.3.attn.qkv.weight": {
3304
+ "dtype": "F32",
3305
+ "shape": [
3306
+ 2304,
3307
+ 768
3308
+ ]
3309
+ },
3310
+ "vision.left_wrist_rgb.blocks.3.mlp.fc1.bias": {
3311
+ "dtype": "F32",
3312
+ "shape": [
3313
+ 3072
3314
+ ]
3315
+ },
3316
+ "vision.left_wrist_rgb.blocks.3.mlp.fc1.weight": {
3317
+ "dtype": "F32",
3318
+ "shape": [
3319
+ 3072,
3320
+ 768
3321
+ ]
3322
+ },
3323
+ "vision.left_wrist_rgb.blocks.3.mlp.fc2.bias": {
3324
+ "dtype": "F32",
3325
+ "shape": [
3326
+ 768
3327
+ ]
3328
+ },
3329
+ "vision.left_wrist_rgb.blocks.3.mlp.fc2.weight": {
3330
+ "dtype": "F32",
3331
+ "shape": [
3332
+ 768,
3333
+ 3072
3334
+ ]
3335
+ },
3336
+ "vision.left_wrist_rgb.blocks.3.norm1.bias": {
3337
+ "dtype": "F32",
3338
+ "shape": [
3339
+ 768
3340
+ ]
3341
+ },
3342
+ "vision.left_wrist_rgb.blocks.3.norm1.weight": {
3343
+ "dtype": "F32",
3344
+ "shape": [
3345
+ 768
3346
+ ]
3347
+ },
3348
+ "vision.left_wrist_rgb.blocks.3.norm2.bias": {
3349
+ "dtype": "F32",
3350
+ "shape": [
3351
+ 768
3352
+ ]
3353
+ },
3354
+ "vision.left_wrist_rgb.blocks.3.norm2.weight": {
3355
+ "dtype": "F32",
3356
+ "shape": [
3357
+ 768
3358
+ ]
3359
+ },
3360
+ "vision.left_wrist_rgb.blocks.4.attn.proj.bias": {
3361
+ "dtype": "F32",
3362
+ "shape": [
3363
+ 768
3364
+ ]
3365
+ },
3366
+ "vision.left_wrist_rgb.blocks.4.attn.proj.weight": {
3367
+ "dtype": "F32",
3368
+ "shape": [
3369
+ 768,
3370
+ 768
3371
+ ]
3372
+ },
3373
+ "vision.left_wrist_rgb.blocks.4.attn.qkv.bias": {
3374
+ "dtype": "F32",
3375
+ "shape": [
3376
+ 2304
3377
+ ]
3378
+ },
3379
+ "vision.left_wrist_rgb.blocks.4.attn.qkv.weight": {
3380
+ "dtype": "F32",
3381
+ "shape": [
3382
+ 2304,
3383
+ 768
3384
+ ]
3385
+ },
3386
+ "vision.left_wrist_rgb.blocks.4.mlp.fc1.bias": {
3387
+ "dtype": "F32",
3388
+ "shape": [
3389
+ 3072
3390
+ ]
3391
+ },
3392
+ "vision.left_wrist_rgb.blocks.4.mlp.fc1.weight": {
3393
+ "dtype": "F32",
3394
+ "shape": [
3395
+ 3072,
3396
+ 768
3397
+ ]
3398
+ },
3399
+ "vision.left_wrist_rgb.blocks.4.mlp.fc2.bias": {
3400
+ "dtype": "F32",
3401
+ "shape": [
3402
+ 768
3403
+ ]
3404
+ },
3405
+ "vision.left_wrist_rgb.blocks.4.mlp.fc2.weight": {
3406
+ "dtype": "F32",
3407
+ "shape": [
3408
+ 768,
3409
+ 3072
3410
+ ]
3411
+ },
3412
+ "vision.left_wrist_rgb.blocks.4.norm1.bias": {
3413
+ "dtype": "F32",
3414
+ "shape": [
3415
+ 768
3416
+ ]
3417
+ },
3418
+ "vision.left_wrist_rgb.blocks.4.norm1.weight": {
3419
+ "dtype": "F32",
3420
+ "shape": [
3421
+ 768
3422
+ ]
3423
+ },
3424
+ "vision.left_wrist_rgb.blocks.4.norm2.bias": {
3425
+ "dtype": "F32",
3426
+ "shape": [
3427
+ 768
3428
+ ]
3429
+ },
3430
+ "vision.left_wrist_rgb.blocks.4.norm2.weight": {
3431
+ "dtype": "F32",
3432
+ "shape": [
3433
+ 768
3434
+ ]
3435
+ },
3436
+ "vision.left_wrist_rgb.blocks.5.attn.proj.bias": {
3437
+ "dtype": "F32",
3438
+ "shape": [
3439
+ 768
3440
+ ]
3441
+ },
3442
+ "vision.left_wrist_rgb.blocks.5.attn.proj.weight": {
3443
+ "dtype": "F32",
3444
+ "shape": [
3445
+ 768,
3446
+ 768
3447
+ ]
3448
+ },
3449
+ "vision.left_wrist_rgb.blocks.5.attn.qkv.bias": {
3450
+ "dtype": "F32",
3451
+ "shape": [
3452
+ 2304
3453
+ ]
3454
+ },
3455
+ "vision.left_wrist_rgb.blocks.5.attn.qkv.weight": {
3456
+ "dtype": "F32",
3457
+ "shape": [
3458
+ 2304,
3459
+ 768
3460
+ ]
3461
+ },
3462
+ "vision.left_wrist_rgb.blocks.5.mlp.fc1.bias": {
3463
+ "dtype": "F32",
3464
+ "shape": [
3465
+ 3072
3466
+ ]
3467
+ },
3468
+ "vision.left_wrist_rgb.blocks.5.mlp.fc1.weight": {
3469
+ "dtype": "F32",
3470
+ "shape": [
3471
+ 3072,
3472
+ 768
3473
+ ]
3474
+ },
3475
+ "vision.left_wrist_rgb.blocks.5.mlp.fc2.bias": {
3476
+ "dtype": "F32",
3477
+ "shape": [
3478
+ 768
3479
+ ]
3480
+ },
3481
+ "vision.left_wrist_rgb.blocks.5.mlp.fc2.weight": {
3482
+ "dtype": "F32",
3483
+ "shape": [
3484
+ 768,
3485
+ 3072
3486
+ ]
3487
+ },
3488
+ "vision.left_wrist_rgb.blocks.5.norm1.bias": {
3489
+ "dtype": "F32",
3490
+ "shape": [
3491
+ 768
3492
+ ]
3493
+ },
3494
+ "vision.left_wrist_rgb.blocks.5.norm1.weight": {
3495
+ "dtype": "F32",
3496
+ "shape": [
3497
+ 768
3498
+ ]
3499
+ },
3500
+ "vision.left_wrist_rgb.blocks.5.norm2.bias": {
3501
+ "dtype": "F32",
3502
+ "shape": [
3503
+ 768
3504
+ ]
3505
+ },
3506
+ "vision.left_wrist_rgb.blocks.5.norm2.weight": {
3507
+ "dtype": "F32",
3508
+ "shape": [
3509
+ 768
3510
+ ]
3511
+ },
3512
+ "vision.left_wrist_rgb.blocks.6.attn.proj.bias": {
3513
+ "dtype": "F32",
3514
+ "shape": [
3515
+ 768
3516
+ ]
3517
+ },
3518
+ "vision.left_wrist_rgb.blocks.6.attn.proj.weight": {
3519
+ "dtype": "F32",
3520
+ "shape": [
3521
+ 768,
3522
+ 768
3523
+ ]
3524
+ },
3525
+ "vision.left_wrist_rgb.blocks.6.attn.qkv.bias": {
3526
+ "dtype": "F32",
3527
+ "shape": [
3528
+ 2304
3529
+ ]
3530
+ },
3531
+ "vision.left_wrist_rgb.blocks.6.attn.qkv.weight": {
3532
+ "dtype": "F32",
3533
+ "shape": [
3534
+ 2304,
3535
+ 768
3536
+ ]
3537
+ },
3538
+ "vision.left_wrist_rgb.blocks.6.mlp.fc1.bias": {
3539
+ "dtype": "F32",
3540
+ "shape": [
3541
+ 3072
3542
+ ]
3543
+ },
3544
+ "vision.left_wrist_rgb.blocks.6.mlp.fc1.weight": {
3545
+ "dtype": "F32",
3546
+ "shape": [
3547
+ 3072,
3548
+ 768
3549
+ ]
3550
+ },
3551
+ "vision.left_wrist_rgb.blocks.6.mlp.fc2.bias": {
3552
+ "dtype": "F32",
3553
+ "shape": [
3554
+ 768
3555
+ ]
3556
+ },
3557
+ "vision.left_wrist_rgb.blocks.6.mlp.fc2.weight": {
3558
+ "dtype": "F32",
3559
+ "shape": [
3560
+ 768,
3561
+ 3072
3562
+ ]
3563
+ },
3564
+ "vision.left_wrist_rgb.blocks.6.norm1.bias": {
3565
+ "dtype": "F32",
3566
+ "shape": [
3567
+ 768
3568
+ ]
3569
+ },
3570
+ "vision.left_wrist_rgb.blocks.6.norm1.weight": {
3571
+ "dtype": "F32",
3572
+ "shape": [
3573
+ 768
3574
+ ]
3575
+ },
3576
+ "vision.left_wrist_rgb.blocks.6.norm2.bias": {
3577
+ "dtype": "F32",
3578
+ "shape": [
3579
+ 768
3580
+ ]
3581
+ },
3582
+ "vision.left_wrist_rgb.blocks.6.norm2.weight": {
3583
+ "dtype": "F32",
3584
+ "shape": [
3585
+ 768
3586
+ ]
3587
+ },
3588
+ "vision.left_wrist_rgb.blocks.7.attn.proj.bias": {
3589
+ "dtype": "F32",
3590
+ "shape": [
3591
+ 768
3592
+ ]
3593
+ },
3594
+ "vision.left_wrist_rgb.blocks.7.attn.proj.weight": {
3595
+ "dtype": "F32",
3596
+ "shape": [
3597
+ 768,
3598
+ 768
3599
+ ]
3600
+ },
3601
+ "vision.left_wrist_rgb.blocks.7.attn.qkv.bias": {
3602
+ "dtype": "F32",
3603
+ "shape": [
3604
+ 2304
3605
+ ]
3606
+ },
3607
+ "vision.left_wrist_rgb.blocks.7.attn.qkv.weight": {
3608
+ "dtype": "F32",
3609
+ "shape": [
3610
+ 2304,
3611
+ 768
3612
+ ]
3613
+ },
3614
+ "vision.left_wrist_rgb.blocks.7.mlp.fc1.bias": {
3615
+ "dtype": "F32",
3616
+ "shape": [
3617
+ 3072
3618
+ ]
3619
+ },
3620
+ "vision.left_wrist_rgb.blocks.7.mlp.fc1.weight": {
3621
+ "dtype": "F32",
3622
+ "shape": [
3623
+ 3072,
3624
+ 768
3625
+ ]
3626
+ },
3627
+ "vision.left_wrist_rgb.blocks.7.mlp.fc2.bias": {
3628
+ "dtype": "F32",
3629
+ "shape": [
3630
+ 768
3631
+ ]
3632
+ },
3633
+ "vision.left_wrist_rgb.blocks.7.mlp.fc2.weight": {
3634
+ "dtype": "F32",
3635
+ "shape": [
3636
+ 768,
3637
+ 3072
3638
+ ]
3639
+ },
3640
+ "vision.left_wrist_rgb.blocks.7.norm1.bias": {
3641
+ "dtype": "F32",
3642
+ "shape": [
3643
+ 768
3644
+ ]
3645
+ },
3646
+ "vision.left_wrist_rgb.blocks.7.norm1.weight": {
3647
+ "dtype": "F32",
3648
+ "shape": [
3649
+ 768
3650
+ ]
3651
+ },
3652
+ "vision.left_wrist_rgb.blocks.7.norm2.bias": {
3653
+ "dtype": "F32",
3654
+ "shape": [
3655
+ 768
3656
+ ]
3657
+ },
3658
+ "vision.left_wrist_rgb.blocks.7.norm2.weight": {
3659
+ "dtype": "F32",
3660
+ "shape": [
3661
+ 768
3662
+ ]
3663
+ },
3664
+ "vision.left_wrist_rgb.blocks.8.attn.proj.bias": {
3665
+ "dtype": "F32",
3666
+ "shape": [
3667
+ 768
3668
+ ]
3669
+ },
3670
+ "vision.left_wrist_rgb.blocks.8.attn.proj.weight": {
3671
+ "dtype": "F32",
3672
+ "shape": [
3673
+ 768,
3674
+ 768
3675
+ ]
3676
+ },
3677
+ "vision.left_wrist_rgb.blocks.8.attn.qkv.bias": {
3678
+ "dtype": "F32",
3679
+ "shape": [
3680
+ 2304
3681
+ ]
3682
+ },
3683
+ "vision.left_wrist_rgb.blocks.8.attn.qkv.weight": {
3684
+ "dtype": "F32",
3685
+ "shape": [
3686
+ 2304,
3687
+ 768
3688
+ ]
3689
+ },
3690
+ "vision.left_wrist_rgb.blocks.8.mlp.fc1.bias": {
3691
+ "dtype": "F32",
3692
+ "shape": [
3693
+ 3072
3694
+ ]
3695
+ },
3696
+ "vision.left_wrist_rgb.blocks.8.mlp.fc1.weight": {
3697
+ "dtype": "F32",
3698
+ "shape": [
3699
+ 3072,
3700
+ 768
3701
+ ]
3702
+ },
3703
+ "vision.left_wrist_rgb.blocks.8.mlp.fc2.bias": {
3704
+ "dtype": "F32",
3705
+ "shape": [
3706
+ 768
3707
+ ]
3708
+ },
3709
+ "vision.left_wrist_rgb.blocks.8.mlp.fc2.weight": {
3710
+ "dtype": "F32",
3711
+ "shape": [
3712
+ 768,
3713
+ 3072
3714
+ ]
3715
+ },
3716
+ "vision.left_wrist_rgb.blocks.8.norm1.bias": {
3717
+ "dtype": "F32",
3718
+ "shape": [
3719
+ 768
3720
+ ]
3721
+ },
3722
+ "vision.left_wrist_rgb.blocks.8.norm1.weight": {
3723
+ "dtype": "F32",
3724
+ "shape": [
3725
+ 768
3726
+ ]
3727
+ },
3728
+ "vision.left_wrist_rgb.blocks.8.norm2.bias": {
3729
+ "dtype": "F32",
3730
+ "shape": [
3731
+ 768
3732
+ ]
3733
+ },
3734
+ "vision.left_wrist_rgb.blocks.8.norm2.weight": {
3735
+ "dtype": "F32",
3736
+ "shape": [
3737
+ 768
3738
+ ]
3739
+ },
3740
+ "vision.left_wrist_rgb.blocks.9.attn.proj.bias": {
3741
+ "dtype": "F32",
3742
+ "shape": [
3743
+ 768
3744
+ ]
3745
+ },
3746
+ "vision.left_wrist_rgb.blocks.9.attn.proj.weight": {
3747
+ "dtype": "F32",
3748
+ "shape": [
3749
+ 768,
3750
+ 768
3751
+ ]
3752
+ },
3753
+ "vision.left_wrist_rgb.blocks.9.attn.qkv.bias": {
3754
+ "dtype": "F32",
3755
+ "shape": [
3756
+ 2304
3757
+ ]
3758
+ },
3759
+ "vision.left_wrist_rgb.blocks.9.attn.qkv.weight": {
3760
+ "dtype": "F32",
3761
+ "shape": [
3762
+ 2304,
3763
+ 768
3764
+ ]
3765
+ },
3766
+ "vision.left_wrist_rgb.blocks.9.mlp.fc1.bias": {
3767
+ "dtype": "F32",
3768
+ "shape": [
3769
+ 3072
3770
+ ]
3771
+ },
3772
+ "vision.left_wrist_rgb.blocks.9.mlp.fc1.weight": {
3773
+ "dtype": "F32",
3774
+ "shape": [
3775
+ 3072,
3776
+ 768
3777
+ ]
3778
+ },
3779
+ "vision.left_wrist_rgb.blocks.9.mlp.fc2.bias": {
3780
+ "dtype": "F32",
3781
+ "shape": [
3782
+ 768
3783
+ ]
3784
+ },
3785
+ "vision.left_wrist_rgb.blocks.9.mlp.fc2.weight": {
3786
+ "dtype": "F32",
3787
+ "shape": [
3788
+ 768,
3789
+ 3072
3790
+ ]
3791
+ },
3792
+ "vision.left_wrist_rgb.blocks.9.norm1.bias": {
3793
+ "dtype": "F32",
3794
+ "shape": [
3795
+ 768
3796
+ ]
3797
+ },
3798
+ "vision.left_wrist_rgb.blocks.9.norm1.weight": {
3799
+ "dtype": "F32",
3800
+ "shape": [
3801
+ 768
3802
+ ]
3803
+ },
3804
+ "vision.left_wrist_rgb.blocks.9.norm2.bias": {
3805
+ "dtype": "F32",
3806
+ "shape": [
3807
+ 768
3808
+ ]
3809
+ },
3810
+ "vision.left_wrist_rgb.blocks.9.norm2.weight": {
3811
+ "dtype": "F32",
3812
+ "shape": [
3813
+ 768
3814
+ ]
3815
+ },
3816
+ "vision.left_wrist_rgb.cls_token": {
3817
+ "dtype": "F32",
3818
+ "shape": [
3819
+ 1,
3820
+ 1,
3821
+ 768
3822
+ ]
3823
+ },
3824
+ "vision.left_wrist_rgb.norm.bias": {
3825
+ "dtype": "F32",
3826
+ "shape": [
3827
+ 768
3828
+ ]
3829
+ },
3830
+ "vision.left_wrist_rgb.norm.weight": {
3831
+ "dtype": "F32",
3832
+ "shape": [
3833
+ 768
3834
+ ]
3835
+ },
3836
+ "vision.left_wrist_rgb.norm_pre.bias": {
3837
+ "dtype": "F32",
3838
+ "shape": [
3839
+ 768
3840
+ ]
3841
+ },
3842
+ "vision.left_wrist_rgb.norm_pre.weight": {
3843
+ "dtype": "F32",
3844
+ "shape": [
3845
+ 768
3846
+ ]
3847
+ },
3848
+ "vision.left_wrist_rgb.patch_embed.proj.weight": {
3849
+ "dtype": "F32",
3850
+ "shape": [
3851
+ 768,
3852
+ 3,
3853
+ 16,
3854
+ 16
3855
+ ]
3856
+ },
3857
+ "vision.left_wrist_rgb.pos_embed": {
3858
+ "dtype": "F32",
3859
+ "shape": [
3860
+ 1,
3861
+ 197,
3862
+ 768
3863
+ ]
3864
+ },
3865
+ "vision.right_wrist_rgb.blocks.0.attn.proj.bias": {
3866
+ "dtype": "F32",
3867
+ "shape": [
3868
+ 768
3869
+ ]
3870
+ },
3871
+ "vision.right_wrist_rgb.blocks.0.attn.proj.weight": {
3872
+ "dtype": "F32",
3873
+ "shape": [
3874
+ 768,
3875
+ 768
3876
+ ]
3877
+ },
3878
+ "vision.right_wrist_rgb.blocks.0.attn.qkv.bias": {
3879
+ "dtype": "F32",
3880
+ "shape": [
3881
+ 2304
3882
+ ]
3883
+ },
3884
+ "vision.right_wrist_rgb.blocks.0.attn.qkv.weight": {
3885
+ "dtype": "F32",
3886
+ "shape": [
3887
+ 2304,
3888
+ 768
3889
+ ]
3890
+ },
3891
+ "vision.right_wrist_rgb.blocks.0.mlp.fc1.bias": {
3892
+ "dtype": "F32",
3893
+ "shape": [
3894
+ 3072
3895
+ ]
3896
+ },
3897
+ "vision.right_wrist_rgb.blocks.0.mlp.fc1.weight": {
3898
+ "dtype": "F32",
3899
+ "shape": [
3900
+ 3072,
3901
+ 768
3902
+ ]
3903
+ },
3904
+ "vision.right_wrist_rgb.blocks.0.mlp.fc2.bias": {
3905
+ "dtype": "F32",
3906
+ "shape": [
3907
+ 768
3908
+ ]
3909
+ },
3910
+ "vision.right_wrist_rgb.blocks.0.mlp.fc2.weight": {
3911
+ "dtype": "F32",
3912
+ "shape": [
3913
+ 768,
3914
+ 3072
3915
+ ]
3916
+ },
3917
+ "vision.right_wrist_rgb.blocks.0.norm1.bias": {
3918
+ "dtype": "F32",
3919
+ "shape": [
3920
+ 768
3921
+ ]
3922
+ },
3923
+ "vision.right_wrist_rgb.blocks.0.norm1.weight": {
3924
+ "dtype": "F32",
3925
+ "shape": [
3926
+ 768
3927
+ ]
3928
+ },
3929
+ "vision.right_wrist_rgb.blocks.0.norm2.bias": {
3930
+ "dtype": "F32",
3931
+ "shape": [
3932
+ 768
3933
+ ]
3934
+ },
3935
+ "vision.right_wrist_rgb.blocks.0.norm2.weight": {
3936
+ "dtype": "F32",
3937
+ "shape": [
3938
+ 768
3939
+ ]
3940
+ },
3941
+ "vision.right_wrist_rgb.blocks.1.attn.proj.bias": {
3942
+ "dtype": "F32",
3943
+ "shape": [
3944
+ 768
3945
+ ]
3946
+ },
3947
+ "vision.right_wrist_rgb.blocks.1.attn.proj.weight": {
3948
+ "dtype": "F32",
3949
+ "shape": [
3950
+ 768,
3951
+ 768
3952
+ ]
3953
+ },
3954
+ "vision.right_wrist_rgb.blocks.1.attn.qkv.bias": {
3955
+ "dtype": "F32",
3956
+ "shape": [
3957
+ 2304
3958
+ ]
3959
+ },
3960
+ "vision.right_wrist_rgb.blocks.1.attn.qkv.weight": {
3961
+ "dtype": "F32",
3962
+ "shape": [
3963
+ 2304,
3964
+ 768
3965
+ ]
3966
+ },
3967
+ "vision.right_wrist_rgb.blocks.1.mlp.fc1.bias": {
3968
+ "dtype": "F32",
3969
+ "shape": [
3970
+ 3072
3971
+ ]
3972
+ },
3973
+ "vision.right_wrist_rgb.blocks.1.mlp.fc1.weight": {
3974
+ "dtype": "F32",
3975
+ "shape": [
3976
+ 3072,
3977
+ 768
3978
+ ]
3979
+ },
3980
+ "vision.right_wrist_rgb.blocks.1.mlp.fc2.bias": {
3981
+ "dtype": "F32",
3982
+ "shape": [
3983
+ 768
3984
+ ]
3985
+ },
3986
+ "vision.right_wrist_rgb.blocks.1.mlp.fc2.weight": {
3987
+ "dtype": "F32",
3988
+ "shape": [
3989
+ 768,
3990
+ 3072
3991
+ ]
3992
+ },
3993
+ "vision.right_wrist_rgb.blocks.1.norm1.bias": {
3994
+ "dtype": "F32",
3995
+ "shape": [
3996
+ 768
3997
+ ]
3998
+ },
3999
+ "vision.right_wrist_rgb.blocks.1.norm1.weight": {
4000
+ "dtype": "F32",
4001
+ "shape": [
4002
+ 768
4003
+ ]
4004
+ },
4005
+ "vision.right_wrist_rgb.blocks.1.norm2.bias": {
4006
+ "dtype": "F32",
4007
+ "shape": [
4008
+ 768
4009
+ ]
4010
+ },
4011
+ "vision.right_wrist_rgb.blocks.1.norm2.weight": {
4012
+ "dtype": "F32",
4013
+ "shape": [
4014
+ 768
4015
+ ]
4016
+ },
4017
+ "vision.right_wrist_rgb.blocks.10.attn.proj.bias": {
4018
+ "dtype": "F32",
4019
+ "shape": [
4020
+ 768
4021
+ ]
4022
+ },
4023
+ "vision.right_wrist_rgb.blocks.10.attn.proj.weight": {
4024
+ "dtype": "F32",
4025
+ "shape": [
4026
+ 768,
4027
+ 768
4028
+ ]
4029
+ },
4030
+ "vision.right_wrist_rgb.blocks.10.attn.qkv.bias": {
4031
+ "dtype": "F32",
4032
+ "shape": [
4033
+ 2304
4034
+ ]
4035
+ },
4036
+ "vision.right_wrist_rgb.blocks.10.attn.qkv.weight": {
4037
+ "dtype": "F32",
4038
+ "shape": [
4039
+ 2304,
4040
+ 768
4041
+ ]
4042
+ },
4043
+ "vision.right_wrist_rgb.blocks.10.mlp.fc1.bias": {
4044
+ "dtype": "F32",
4045
+ "shape": [
4046
+ 3072
4047
+ ]
4048
+ },
4049
+ "vision.right_wrist_rgb.blocks.10.mlp.fc1.weight": {
4050
+ "dtype": "F32",
4051
+ "shape": [
4052
+ 3072,
4053
+ 768
4054
+ ]
4055
+ },
4056
+ "vision.right_wrist_rgb.blocks.10.mlp.fc2.bias": {
4057
+ "dtype": "F32",
4058
+ "shape": [
4059
+ 768
4060
+ ]
4061
+ },
4062
+ "vision.right_wrist_rgb.blocks.10.mlp.fc2.weight": {
4063
+ "dtype": "F32",
4064
+ "shape": [
4065
+ 768,
4066
+ 3072
4067
+ ]
4068
+ },
4069
+ "vision.right_wrist_rgb.blocks.10.norm1.bias": {
4070
+ "dtype": "F32",
4071
+ "shape": [
4072
+ 768
4073
+ ]
4074
+ },
4075
+ "vision.right_wrist_rgb.blocks.10.norm1.weight": {
4076
+ "dtype": "F32",
4077
+ "shape": [
4078
+ 768
4079
+ ]
4080
+ },
4081
+ "vision.right_wrist_rgb.blocks.10.norm2.bias": {
4082
+ "dtype": "F32",
4083
+ "shape": [
4084
+ 768
4085
+ ]
4086
+ },
4087
+ "vision.right_wrist_rgb.blocks.10.norm2.weight": {
4088
+ "dtype": "F32",
4089
+ "shape": [
4090
+ 768
4091
+ ]
4092
+ },
4093
+ "vision.right_wrist_rgb.blocks.11.attn.proj.bias": {
4094
+ "dtype": "F32",
4095
+ "shape": [
4096
+ 768
4097
+ ]
4098
+ },
4099
+ "vision.right_wrist_rgb.blocks.11.attn.proj.weight": {
4100
+ "dtype": "F32",
4101
+ "shape": [
4102
+ 768,
4103
+ 768
4104
+ ]
4105
+ },
4106
+ "vision.right_wrist_rgb.blocks.11.attn.qkv.bias": {
4107
+ "dtype": "F32",
4108
+ "shape": [
4109
+ 2304
4110
+ ]
4111
+ },
4112
+ "vision.right_wrist_rgb.blocks.11.attn.qkv.weight": {
4113
+ "dtype": "F32",
4114
+ "shape": [
4115
+ 2304,
4116
+ 768
4117
+ ]
4118
+ },
4119
+ "vision.right_wrist_rgb.blocks.11.mlp.fc1.bias": {
4120
+ "dtype": "F32",
4121
+ "shape": [
4122
+ 3072
4123
+ ]
4124
+ },
4125
+ "vision.right_wrist_rgb.blocks.11.mlp.fc1.weight": {
4126
+ "dtype": "F32",
4127
+ "shape": [
4128
+ 3072,
4129
+ 768
4130
+ ]
4131
+ },
4132
+ "vision.right_wrist_rgb.blocks.11.mlp.fc2.bias": {
4133
+ "dtype": "F32",
4134
+ "shape": [
4135
+ 768
4136
+ ]
4137
+ },
4138
+ "vision.right_wrist_rgb.blocks.11.mlp.fc2.weight": {
4139
+ "dtype": "F32",
4140
+ "shape": [
4141
+ 768,
4142
+ 3072
4143
+ ]
4144
+ },
4145
+ "vision.right_wrist_rgb.blocks.11.norm1.bias": {
4146
+ "dtype": "F32",
4147
+ "shape": [
4148
+ 768
4149
+ ]
4150
+ },
4151
+ "vision.right_wrist_rgb.blocks.11.norm1.weight": {
4152
+ "dtype": "F32",
4153
+ "shape": [
4154
+ 768
4155
+ ]
4156
+ },
4157
+ "vision.right_wrist_rgb.blocks.11.norm2.bias": {
4158
+ "dtype": "F32",
4159
+ "shape": [
4160
+ 768
4161
+ ]
4162
+ },
4163
+ "vision.right_wrist_rgb.blocks.11.norm2.weight": {
4164
+ "dtype": "F32",
4165
+ "shape": [
4166
+ 768
4167
+ ]
4168
+ },
4169
+ "vision.right_wrist_rgb.blocks.2.attn.proj.bias": {
4170
+ "dtype": "F32",
4171
+ "shape": [
4172
+ 768
4173
+ ]
4174
+ },
4175
+ "vision.right_wrist_rgb.blocks.2.attn.proj.weight": {
4176
+ "dtype": "F32",
4177
+ "shape": [
4178
+ 768,
4179
+ 768
4180
+ ]
4181
+ },
4182
+ "vision.right_wrist_rgb.blocks.2.attn.qkv.bias": {
4183
+ "dtype": "F32",
4184
+ "shape": [
4185
+ 2304
4186
+ ]
4187
+ },
4188
+ "vision.right_wrist_rgb.blocks.2.attn.qkv.weight": {
4189
+ "dtype": "F32",
4190
+ "shape": [
4191
+ 2304,
4192
+ 768
4193
+ ]
4194
+ },
4195
+ "vision.right_wrist_rgb.blocks.2.mlp.fc1.bias": {
4196
+ "dtype": "F32",
4197
+ "shape": [
4198
+ 3072
4199
+ ]
4200
+ },
4201
+ "vision.right_wrist_rgb.blocks.2.mlp.fc1.weight": {
4202
+ "dtype": "F32",
4203
+ "shape": [
4204
+ 3072,
4205
+ 768
4206
+ ]
4207
+ },
4208
+ "vision.right_wrist_rgb.blocks.2.mlp.fc2.bias": {
4209
+ "dtype": "F32",
4210
+ "shape": [
4211
+ 768
4212
+ ]
4213
+ },
4214
+ "vision.right_wrist_rgb.blocks.2.mlp.fc2.weight": {
4215
+ "dtype": "F32",
4216
+ "shape": [
4217
+ 768,
4218
+ 3072
4219
+ ]
4220
+ },
4221
+ "vision.right_wrist_rgb.blocks.2.norm1.bias": {
4222
+ "dtype": "F32",
4223
+ "shape": [
4224
+ 768
4225
+ ]
4226
+ },
4227
+ "vision.right_wrist_rgb.blocks.2.norm1.weight": {
4228
+ "dtype": "F32",
4229
+ "shape": [
4230
+ 768
4231
+ ]
4232
+ },
4233
+ "vision.right_wrist_rgb.blocks.2.norm2.bias": {
4234
+ "dtype": "F32",
4235
+ "shape": [
4236
+ 768
4237
+ ]
4238
+ },
4239
+ "vision.right_wrist_rgb.blocks.2.norm2.weight": {
4240
+ "dtype": "F32",
4241
+ "shape": [
4242
+ 768
4243
+ ]
4244
+ },
4245
+ "vision.right_wrist_rgb.blocks.3.attn.proj.bias": {
4246
+ "dtype": "F32",
4247
+ "shape": [
4248
+ 768
4249
+ ]
4250
+ },
4251
+ "vision.right_wrist_rgb.blocks.3.attn.proj.weight": {
4252
+ "dtype": "F32",
4253
+ "shape": [
4254
+ 768,
4255
+ 768
4256
+ ]
4257
+ },
4258
+ "vision.right_wrist_rgb.blocks.3.attn.qkv.bias": {
4259
+ "dtype": "F32",
4260
+ "shape": [
4261
+ 2304
4262
+ ]
4263
+ },
4264
+ "vision.right_wrist_rgb.blocks.3.attn.qkv.weight": {
4265
+ "dtype": "F32",
4266
+ "shape": [
4267
+ 2304,
4268
+ 768
4269
+ ]
4270
+ },
4271
+ "vision.right_wrist_rgb.blocks.3.mlp.fc1.bias": {
4272
+ "dtype": "F32",
4273
+ "shape": [
4274
+ 3072
4275
+ ]
4276
+ },
4277
+ "vision.right_wrist_rgb.blocks.3.mlp.fc1.weight": {
4278
+ "dtype": "F32",
4279
+ "shape": [
4280
+ 3072,
4281
+ 768
4282
+ ]
4283
+ },
4284
+ "vision.right_wrist_rgb.blocks.3.mlp.fc2.bias": {
4285
+ "dtype": "F32",
4286
+ "shape": [
4287
+ 768
4288
+ ]
4289
+ },
4290
+ "vision.right_wrist_rgb.blocks.3.mlp.fc2.weight": {
4291
+ "dtype": "F32",
4292
+ "shape": [
4293
+ 768,
4294
+ 3072
4295
+ ]
4296
+ },
4297
+ "vision.right_wrist_rgb.blocks.3.norm1.bias": {
4298
+ "dtype": "F32",
4299
+ "shape": [
4300
+ 768
4301
+ ]
4302
+ },
4303
+ "vision.right_wrist_rgb.blocks.3.norm1.weight": {
4304
+ "dtype": "F32",
4305
+ "shape": [
4306
+ 768
4307
+ ]
4308
+ },
4309
+ "vision.right_wrist_rgb.blocks.3.norm2.bias": {
4310
+ "dtype": "F32",
4311
+ "shape": [
4312
+ 768
4313
+ ]
4314
+ },
4315
+ "vision.right_wrist_rgb.blocks.3.norm2.weight": {
4316
+ "dtype": "F32",
4317
+ "shape": [
4318
+ 768
4319
+ ]
4320
+ },
4321
+ "vision.right_wrist_rgb.blocks.4.attn.proj.bias": {
4322
+ "dtype": "F32",
4323
+ "shape": [
4324
+ 768
4325
+ ]
4326
+ },
4327
+ "vision.right_wrist_rgb.blocks.4.attn.proj.weight": {
4328
+ "dtype": "F32",
4329
+ "shape": [
4330
+ 768,
4331
+ 768
4332
+ ]
4333
+ },
4334
+ "vision.right_wrist_rgb.blocks.4.attn.qkv.bias": {
4335
+ "dtype": "F32",
4336
+ "shape": [
4337
+ 2304
4338
+ ]
4339
+ },
4340
+ "vision.right_wrist_rgb.blocks.4.attn.qkv.weight": {
4341
+ "dtype": "F32",
4342
+ "shape": [
4343
+ 2304,
4344
+ 768
4345
+ ]
4346
+ },
4347
+ "vision.right_wrist_rgb.blocks.4.mlp.fc1.bias": {
4348
+ "dtype": "F32",
4349
+ "shape": [
4350
+ 3072
4351
+ ]
4352
+ },
4353
+ "vision.right_wrist_rgb.blocks.4.mlp.fc1.weight": {
4354
+ "dtype": "F32",
4355
+ "shape": [
4356
+ 3072,
4357
+ 768
4358
+ ]
4359
+ },
4360
+ "vision.right_wrist_rgb.blocks.4.mlp.fc2.bias": {
4361
+ "dtype": "F32",
4362
+ "shape": [
4363
+ 768
4364
+ ]
4365
+ },
4366
+ "vision.right_wrist_rgb.blocks.4.mlp.fc2.weight": {
4367
+ "dtype": "F32",
4368
+ "shape": [
4369
+ 768,
4370
+ 3072
4371
+ ]
4372
+ },
4373
+ "vision.right_wrist_rgb.blocks.4.norm1.bias": {
4374
+ "dtype": "F32",
4375
+ "shape": [
4376
+ 768
4377
+ ]
4378
+ },
4379
+ "vision.right_wrist_rgb.blocks.4.norm1.weight": {
4380
+ "dtype": "F32",
4381
+ "shape": [
4382
+ 768
4383
+ ]
4384
+ },
4385
+ "vision.right_wrist_rgb.blocks.4.norm2.bias": {
4386
+ "dtype": "F32",
4387
+ "shape": [
4388
+ 768
4389
+ ]
4390
+ },
4391
+ "vision.right_wrist_rgb.blocks.4.norm2.weight": {
4392
+ "dtype": "F32",
4393
+ "shape": [
4394
+ 768
4395
+ ]
4396
+ },
4397
+ "vision.right_wrist_rgb.blocks.5.attn.proj.bias": {
4398
+ "dtype": "F32",
4399
+ "shape": [
4400
+ 768
4401
+ ]
4402
+ },
4403
+ "vision.right_wrist_rgb.blocks.5.attn.proj.weight": {
4404
+ "dtype": "F32",
4405
+ "shape": [
4406
+ 768,
4407
+ 768
4408
+ ]
4409
+ },
4410
+ "vision.right_wrist_rgb.blocks.5.attn.qkv.bias": {
4411
+ "dtype": "F32",
4412
+ "shape": [
4413
+ 2304
4414
+ ]
4415
+ },
4416
+ "vision.right_wrist_rgb.blocks.5.attn.qkv.weight": {
4417
+ "dtype": "F32",
4418
+ "shape": [
4419
+ 2304,
4420
+ 768
4421
+ ]
4422
+ },
4423
+ "vision.right_wrist_rgb.blocks.5.mlp.fc1.bias": {
4424
+ "dtype": "F32",
4425
+ "shape": [
4426
+ 3072
4427
+ ]
4428
+ },
4429
+ "vision.right_wrist_rgb.blocks.5.mlp.fc1.weight": {
4430
+ "dtype": "F32",
4431
+ "shape": [
4432
+ 3072,
4433
+ 768
4434
+ ]
4435
+ },
4436
+ "vision.right_wrist_rgb.blocks.5.mlp.fc2.bias": {
4437
+ "dtype": "F32",
4438
+ "shape": [
4439
+ 768
4440
+ ]
4441
+ },
4442
+ "vision.right_wrist_rgb.blocks.5.mlp.fc2.weight": {
4443
+ "dtype": "F32",
4444
+ "shape": [
4445
+ 768,
4446
+ 3072
4447
+ ]
4448
+ },
4449
+ "vision.right_wrist_rgb.blocks.5.norm1.bias": {
4450
+ "dtype": "F32",
4451
+ "shape": [
4452
+ 768
4453
+ ]
4454
+ },
4455
+ "vision.right_wrist_rgb.blocks.5.norm1.weight": {
4456
+ "dtype": "F32",
4457
+ "shape": [
4458
+ 768
4459
+ ]
4460
+ },
4461
+ "vision.right_wrist_rgb.blocks.5.norm2.bias": {
4462
+ "dtype": "F32",
4463
+ "shape": [
4464
+ 768
4465
+ ]
4466
+ },
4467
+ "vision.right_wrist_rgb.blocks.5.norm2.weight": {
4468
+ "dtype": "F32",
4469
+ "shape": [
4470
+ 768
4471
+ ]
4472
+ },
4473
+ "vision.right_wrist_rgb.blocks.6.attn.proj.bias": {
4474
+ "dtype": "F32",
4475
+ "shape": [
4476
+ 768
4477
+ ]
4478
+ },
4479
+ "vision.right_wrist_rgb.blocks.6.attn.proj.weight": {
4480
+ "dtype": "F32",
4481
+ "shape": [
4482
+ 768,
4483
+ 768
4484
+ ]
4485
+ },
4486
+ "vision.right_wrist_rgb.blocks.6.attn.qkv.bias": {
4487
+ "dtype": "F32",
4488
+ "shape": [
4489
+ 2304
4490
+ ]
4491
+ },
4492
+ "vision.right_wrist_rgb.blocks.6.attn.qkv.weight": {
4493
+ "dtype": "F32",
4494
+ "shape": [
4495
+ 2304,
4496
+ 768
4497
+ ]
4498
+ },
4499
+ "vision.right_wrist_rgb.blocks.6.mlp.fc1.bias": {
4500
+ "dtype": "F32",
4501
+ "shape": [
4502
+ 3072
4503
+ ]
4504
+ },
4505
+ "vision.right_wrist_rgb.blocks.6.mlp.fc1.weight": {
4506
+ "dtype": "F32",
4507
+ "shape": [
4508
+ 3072,
4509
+ 768
4510
+ ]
4511
+ },
4512
+ "vision.right_wrist_rgb.blocks.6.mlp.fc2.bias": {
4513
+ "dtype": "F32",
4514
+ "shape": [
4515
+ 768
4516
+ ]
4517
+ },
4518
+ "vision.right_wrist_rgb.blocks.6.mlp.fc2.weight": {
4519
+ "dtype": "F32",
4520
+ "shape": [
4521
+ 768,
4522
+ 3072
4523
+ ]
4524
+ },
4525
+ "vision.right_wrist_rgb.blocks.6.norm1.bias": {
4526
+ "dtype": "F32",
4527
+ "shape": [
4528
+ 768
4529
+ ]
4530
+ },
4531
+ "vision.right_wrist_rgb.blocks.6.norm1.weight": {
4532
+ "dtype": "F32",
4533
+ "shape": [
4534
+ 768
4535
+ ]
4536
+ },
4537
+ "vision.right_wrist_rgb.blocks.6.norm2.bias": {
4538
+ "dtype": "F32",
4539
+ "shape": [
4540
+ 768
4541
+ ]
4542
+ },
4543
+ "vision.right_wrist_rgb.blocks.6.norm2.weight": {
4544
+ "dtype": "F32",
4545
+ "shape": [
4546
+ 768
4547
+ ]
4548
+ },
4549
+ "vision.right_wrist_rgb.blocks.7.attn.proj.bias": {
4550
+ "dtype": "F32",
4551
+ "shape": [
4552
+ 768
4553
+ ]
4554
+ },
4555
+ "vision.right_wrist_rgb.blocks.7.attn.proj.weight": {
4556
+ "dtype": "F32",
4557
+ "shape": [
4558
+ 768,
4559
+ 768
4560
+ ]
4561
+ },
4562
+ "vision.right_wrist_rgb.blocks.7.attn.qkv.bias": {
4563
+ "dtype": "F32",
4564
+ "shape": [
4565
+ 2304
4566
+ ]
4567
+ },
4568
+ "vision.right_wrist_rgb.blocks.7.attn.qkv.weight": {
4569
+ "dtype": "F32",
4570
+ "shape": [
4571
+ 2304,
4572
+ 768
4573
+ ]
4574
+ },
4575
+ "vision.right_wrist_rgb.blocks.7.mlp.fc1.bias": {
4576
+ "dtype": "F32",
4577
+ "shape": [
4578
+ 3072
4579
+ ]
4580
+ },
4581
+ "vision.right_wrist_rgb.blocks.7.mlp.fc1.weight": {
4582
+ "dtype": "F32",
4583
+ "shape": [
4584
+ 3072,
4585
+ 768
4586
+ ]
4587
+ },
4588
+ "vision.right_wrist_rgb.blocks.7.mlp.fc2.bias": {
4589
+ "dtype": "F32",
4590
+ "shape": [
4591
+ 768
4592
+ ]
4593
+ },
4594
+ "vision.right_wrist_rgb.blocks.7.mlp.fc2.weight": {
4595
+ "dtype": "F32",
4596
+ "shape": [
4597
+ 768,
4598
+ 3072
4599
+ ]
4600
+ },
4601
+ "vision.right_wrist_rgb.blocks.7.norm1.bias": {
4602
+ "dtype": "F32",
4603
+ "shape": [
4604
+ 768
4605
+ ]
4606
+ },
4607
+ "vision.right_wrist_rgb.blocks.7.norm1.weight": {
4608
+ "dtype": "F32",
4609
+ "shape": [
4610
+ 768
4611
+ ]
4612
+ },
4613
+ "vision.right_wrist_rgb.blocks.7.norm2.bias": {
4614
+ "dtype": "F32",
4615
+ "shape": [
4616
+ 768
4617
+ ]
4618
+ },
4619
+ "vision.right_wrist_rgb.blocks.7.norm2.weight": {
4620
+ "dtype": "F32",
4621
+ "shape": [
4622
+ 768
4623
+ ]
4624
+ },
4625
+ "vision.right_wrist_rgb.blocks.8.attn.proj.bias": {
4626
+ "dtype": "F32",
4627
+ "shape": [
4628
+ 768
4629
+ ]
4630
+ },
4631
+ "vision.right_wrist_rgb.blocks.8.attn.proj.weight": {
4632
+ "dtype": "F32",
4633
+ "shape": [
4634
+ 768,
4635
+ 768
4636
+ ]
4637
+ },
4638
+ "vision.right_wrist_rgb.blocks.8.attn.qkv.bias": {
4639
+ "dtype": "F32",
4640
+ "shape": [
4641
+ 2304
4642
+ ]
4643
+ },
4644
+ "vision.right_wrist_rgb.blocks.8.attn.qkv.weight": {
4645
+ "dtype": "F32",
4646
+ "shape": [
4647
+ 2304,
4648
+ 768
4649
+ ]
4650
+ },
4651
+ "vision.right_wrist_rgb.blocks.8.mlp.fc1.bias": {
4652
+ "dtype": "F32",
4653
+ "shape": [
4654
+ 3072
4655
+ ]
4656
+ },
4657
+ "vision.right_wrist_rgb.blocks.8.mlp.fc1.weight": {
4658
+ "dtype": "F32",
4659
+ "shape": [
4660
+ 3072,
4661
+ 768
4662
+ ]
4663
+ },
4664
+ "vision.right_wrist_rgb.blocks.8.mlp.fc2.bias": {
4665
+ "dtype": "F32",
4666
+ "shape": [
4667
+ 768
4668
+ ]
4669
+ },
4670
+ "vision.right_wrist_rgb.blocks.8.mlp.fc2.weight": {
4671
+ "dtype": "F32",
4672
+ "shape": [
4673
+ 768,
4674
+ 3072
4675
+ ]
4676
+ },
4677
+ "vision.right_wrist_rgb.blocks.8.norm1.bias": {
4678
+ "dtype": "F32",
4679
+ "shape": [
4680
+ 768
4681
+ ]
4682
+ },
4683
+ "vision.right_wrist_rgb.blocks.8.norm1.weight": {
4684
+ "dtype": "F32",
4685
+ "shape": [
4686
+ 768
4687
+ ]
4688
+ },
4689
+ "vision.right_wrist_rgb.blocks.8.norm2.bias": {
4690
+ "dtype": "F32",
4691
+ "shape": [
4692
+ 768
4693
+ ]
4694
+ },
4695
+ "vision.right_wrist_rgb.blocks.8.norm2.weight": {
4696
+ "dtype": "F32",
4697
+ "shape": [
4698
+ 768
4699
+ ]
4700
+ },
4701
+ "vision.right_wrist_rgb.blocks.9.attn.proj.bias": {
4702
+ "dtype": "F32",
4703
+ "shape": [
4704
+ 768
4705
+ ]
4706
+ },
4707
+ "vision.right_wrist_rgb.blocks.9.attn.proj.weight": {
4708
+ "dtype": "F32",
4709
+ "shape": [
4710
+ 768,
4711
+ 768
4712
+ ]
4713
+ },
4714
+ "vision.right_wrist_rgb.blocks.9.attn.qkv.bias": {
4715
+ "dtype": "F32",
4716
+ "shape": [
4717
+ 2304
4718
+ ]
4719
+ },
4720
+ "vision.right_wrist_rgb.blocks.9.attn.qkv.weight": {
4721
+ "dtype": "F32",
4722
+ "shape": [
4723
+ 2304,
4724
+ 768
4725
+ ]
4726
+ },
4727
+ "vision.right_wrist_rgb.blocks.9.mlp.fc1.bias": {
4728
+ "dtype": "F32",
4729
+ "shape": [
4730
+ 3072
4731
+ ]
4732
+ },
4733
+ "vision.right_wrist_rgb.blocks.9.mlp.fc1.weight": {
4734
+ "dtype": "F32",
4735
+ "shape": [
4736
+ 3072,
4737
+ 768
4738
+ ]
4739
+ },
4740
+ "vision.right_wrist_rgb.blocks.9.mlp.fc2.bias": {
4741
+ "dtype": "F32",
4742
+ "shape": [
4743
+ 768
4744
+ ]
4745
+ },
4746
+ "vision.right_wrist_rgb.blocks.9.mlp.fc2.weight": {
4747
+ "dtype": "F32",
4748
+ "shape": [
4749
+ 768,
4750
+ 3072
4751
+ ]
4752
+ },
4753
+ "vision.right_wrist_rgb.blocks.9.norm1.bias": {
4754
+ "dtype": "F32",
4755
+ "shape": [
4756
+ 768
4757
+ ]
4758
+ },
4759
+ "vision.right_wrist_rgb.blocks.9.norm1.weight": {
4760
+ "dtype": "F32",
4761
+ "shape": [
4762
+ 768
4763
+ ]
4764
+ },
4765
+ "vision.right_wrist_rgb.blocks.9.norm2.bias": {
4766
+ "dtype": "F32",
4767
+ "shape": [
4768
+ 768
4769
+ ]
4770
+ },
4771
+ "vision.right_wrist_rgb.blocks.9.norm2.weight": {
4772
+ "dtype": "F32",
4773
+ "shape": [
4774
+ 768
4775
+ ]
4776
+ },
4777
+ "vision.right_wrist_rgb.cls_token": {
4778
+ "dtype": "F32",
4779
+ "shape": [
4780
+ 1,
4781
+ 1,
4782
+ 768
4783
+ ]
4784
+ },
4785
+ "vision.right_wrist_rgb.norm.bias": {
4786
+ "dtype": "F32",
4787
+ "shape": [
4788
+ 768
4789
+ ]
4790
+ },
4791
+ "vision.right_wrist_rgb.norm.weight": {
4792
+ "dtype": "F32",
4793
+ "shape": [
4794
+ 768
4795
+ ]
4796
+ },
4797
+ "vision.right_wrist_rgb.norm_pre.bias": {
4798
+ "dtype": "F32",
4799
+ "shape": [
4800
+ 768
4801
+ ]
4802
+ },
4803
+ "vision.right_wrist_rgb.norm_pre.weight": {
4804
+ "dtype": "F32",
4805
+ "shape": [
4806
+ 768
4807
+ ]
4808
+ },
4809
+ "vision.right_wrist_rgb.patch_embed.proj.weight": {
4810
+ "dtype": "F32",
4811
+ "shape": [
4812
+ 768,
4813
+ 3,
4814
+ 16,
4815
+ 16
4816
+ ]
4817
+ },
4818
+ "vision.right_wrist_rgb.pos_embed": {
4819
+ "dtype": "F32",
4820
+ "shape": [
4821
+ 1,
4822
+ 197,
4823
+ 768
4824
+ ]
4825
+ }
4826
+ }