rchal97 commited on
Commit
9d8147b
·
verified ·
1 Parent(s): d7ef267

Upload 18 files

Browse files
ego_vla_checkpoint/.DS_Store ADDED
Binary file (6.15 kB). View file
 
ego_vla_checkpoint/ckpt-6720/.DS_Store ADDED
Binary file (6.15 kB). View file
 
ego_vla_checkpoint/ckpt-6720/config.json ADDED
@@ -0,0 +1,581 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4",
3
+ "action_output_dim": 48,
4
+ "action_output_ee_2d_dim": 0,
5
+ "action_output_ee_dim": 6,
6
+ "action_output_ee_rot_dim": 12,
7
+ "action_output_hand_dim": 30,
8
+ "action_token_idx": 31733,
9
+ "architectures": [
10
+ "LlavaLlamaModel"
11
+ ],
12
+ "chat_template": null,
13
+ "cvae_hidden_size": 256,
14
+ "cvae_latent_dim": 256,
15
+ "drop_path_rate": 0.0,
16
+ "ee_2d_loss_coeff": 0.0,
17
+ "ee_loss_coeff": 20.0,
18
+ "ee_movement_loss_weight": 1,
19
+ "ee_rot_loss_coeff": 5.0,
20
+ "ee_rot_representation": "rot6d",
21
+ "fps": 0.0,
22
+ "hand_kp_loss_coeff": 0.0,
23
+ "hand_loss_coeff": 5.0,
24
+ "hand_loss_dim": 6,
25
+ "hidden_size": 1536,
26
+ "image_aspect_ratio": "resize",
27
+ "input_placeholder_end_token_idx": 151195,
28
+ "input_placeholder_start_token_idx": 151375,
29
+ "input_placeholder_token_idx": 151375,
30
+ "interpolate_mode": "linear",
31
+ "invalid_token_idx": 151376,
32
+ "invalid_token_weight": 0.1,
33
+ "kl_weight": 0.001,
34
+ "llm_cfg": {
35
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/llm",
36
+ "add_cross_attention": false,
37
+ "architectures": [
38
+ "Qwen2ForCausalLM"
39
+ ],
40
+ "attention_dropout": 0.0,
41
+ "bad_words_ids": null,
42
+ "begin_suppress_tokens": null,
43
+ "bos_token_id": 151643,
44
+ "chunk_size_feed_forward": 0,
45
+ "cross_attention_hidden_size": null,
46
+ "decoder_start_token_id": null,
47
+ "diversity_penalty": 0.0,
48
+ "do_sample": false,
49
+ "early_stopping": false,
50
+ "encoder_no_repeat_ngram_size": 0,
51
+ "eos_token_id": 151645,
52
+ "exponential_decay_length_penalty": null,
53
+ "finetuning_task": null,
54
+ "forced_bos_token_id": null,
55
+ "forced_eos_token_id": null,
56
+ "hidden_act": "silu",
57
+ "hidden_size": 1536,
58
+ "id2label": {
59
+ "0": "LABEL_0",
60
+ "1": "LABEL_1"
61
+ },
62
+ "initializer_range": 0.02,
63
+ "intermediate_size": 8960,
64
+ "is_decoder": false,
65
+ "is_encoder_decoder": false,
66
+ "label2id": {
67
+ "LABEL_0": 0,
68
+ "LABEL_1": 1
69
+ },
70
+ "length_penalty": 1.0,
71
+ "max_length": 20,
72
+ "max_position_embeddings": 32768,
73
+ "max_window_layers": 28,
74
+ "min_length": 0,
75
+ "model_max_length": 4096,
76
+ "model_type": "qwen2",
77
+ "no_repeat_ngram_size": 0,
78
+ "num_attention_heads": 12,
79
+ "num_beam_groups": 1,
80
+ "num_beams": 1,
81
+ "num_hidden_layers": 28,
82
+ "num_key_value_heads": 2,
83
+ "num_return_sequences": 1,
84
+ "output_attentions": false,
85
+ "output_hidden_states": false,
86
+ "output_scores": false,
87
+ "pad_token_id": null,
88
+ "prefix": null,
89
+ "problem_type": null,
90
+ "pruned_heads": {},
91
+ "remove_invalid_values": false,
92
+ "repetition_penalty": 1.0,
93
+ "return_dict": true,
94
+ "return_dict_in_generate": false,
95
+ "rms_norm_eps": 1e-06,
96
+ "rope_theta": 1000000.0,
97
+ "sep_token_id": null,
98
+ "sliding_window": null,
99
+ "suppress_tokens": null,
100
+ "task_specific_params": null,
101
+ "temperature": 1.0,
102
+ "tf_legacy_loss": false,
103
+ "tie_encoder_decoder": false,
104
+ "tie_word_embeddings": true,
105
+ "tokenizer_class": null,
106
+ "tokenizer_model_max_length": 4096,
107
+ "tokenizer_padding_side": "right",
108
+ "top_k": 50,
109
+ "top_p": 1.0,
110
+ "torch_dtype": "bfloat16",
111
+ "torchscript": false,
112
+ "typical_p": 1.0,
113
+ "use_bfloat16": false,
114
+ "use_cache": true,
115
+ "use_sliding_window": false,
116
+ "vocab_size": 151648
117
+ },
118
+ "loss_use_l1": true,
119
+ "merge_hand": true,
120
+ "mm_hidden_size": 1152,
121
+ "mm_projector_cfg": {
122
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/mm_projector",
123
+ "add_cross_attention": false,
124
+ "architectures": [
125
+ "MultimodalProjector"
126
+ ],
127
+ "bad_words_ids": null,
128
+ "begin_suppress_tokens": null,
129
+ "bos_token_id": null,
130
+ "chunk_size_feed_forward": 0,
131
+ "cross_attention_hidden_size": null,
132
+ "decoder_start_token_id": null,
133
+ "diversity_penalty": 0.0,
134
+ "do_sample": false,
135
+ "early_stopping": false,
136
+ "encoder_no_repeat_ngram_size": 0,
137
+ "eos_token_id": null,
138
+ "exponential_decay_length_penalty": null,
139
+ "finetuning_task": null,
140
+ "forced_bos_token_id": null,
141
+ "forced_eos_token_id": null,
142
+ "id2label": {
143
+ "0": "LABEL_0",
144
+ "1": "LABEL_1"
145
+ },
146
+ "is_decoder": false,
147
+ "is_encoder_decoder": false,
148
+ "label2id": {
149
+ "LABEL_0": 0,
150
+ "LABEL_1": 1
151
+ },
152
+ "length_penalty": 1.0,
153
+ "max_length": 20,
154
+ "min_length": 0,
155
+ "mm_projector_type": "mlp_downsample",
156
+ "model_type": "v2l_projector",
157
+ "no_repeat_ngram_size": 0,
158
+ "num_beam_groups": 1,
159
+ "num_beams": 1,
160
+ "num_return_sequences": 1,
161
+ "output_attentions": false,
162
+ "output_hidden_states": false,
163
+ "output_scores": false,
164
+ "pad_token_id": null,
165
+ "prefix": null,
166
+ "problem_type": null,
167
+ "pruned_heads": {},
168
+ "remove_invalid_values": false,
169
+ "repetition_penalty": 1.0,
170
+ "return_dict": true,
171
+ "return_dict_in_generate": false,
172
+ "sep_token_id": null,
173
+ "suppress_tokens": null,
174
+ "task_specific_params": null,
175
+ "temperature": 1.0,
176
+ "tf_legacy_loss": false,
177
+ "tie_encoder_decoder": false,
178
+ "tie_word_embeddings": true,
179
+ "tokenizer_class": null,
180
+ "top_k": 50,
181
+ "top_p": 1.0,
182
+ "torch_dtype": "bfloat16",
183
+ "torchscript": false,
184
+ "typical_p": 1.0,
185
+ "use_bfloat16": false
186
+ },
187
+ "mm_projector_lr": null,
188
+ "mm_use_im_patch_token": true,
189
+ "mm_use_im_start_end": false,
190
+ "mm_vision_select_feature": "cls_patch",
191
+ "mm_vision_select_layer": -2,
192
+ "model_dtype": "torch.bfloat16",
193
+ "model_type": "llava_llama",
194
+ "next_token_loss_coeff": 0.0,
195
+ "num_time_tokens": 0,
196
+ "num_video_frames": 8,
197
+ "proprio_size": 16,
198
+ "resume_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4",
199
+ "s2": false,
200
+ "s2_max_split_size": 336,
201
+ "s2_scales": "336,672,1008",
202
+ "sep_proprio": true,
203
+ "sep_query_token": true,
204
+ "soft_ce_std": 1.0,
205
+ "time_token_format": "<t{t}>",
206
+ "time_token_ids": [],
207
+ "torch_dtype": "bfloat16",
208
+ "traj_decoder_cfg": {
209
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4",
210
+ "add_cross_attention": false,
211
+ "architectures": [
212
+ "LlavaLlamaModel"
213
+ ],
214
+ "bad_words_ids": null,
215
+ "begin_suppress_tokens": null,
216
+ "bos_token_id": null,
217
+ "chunk_size_feed_forward": 0,
218
+ "cross_attention_hidden_size": null,
219
+ "decoder_start_token_id": null,
220
+ "diversity_penalty": 0.0,
221
+ "do_sample": false,
222
+ "early_stopping": false,
223
+ "encoder_no_repeat_ngram_size": 0,
224
+ "eos_token_id": null,
225
+ "exponential_decay_length_penalty": null,
226
+ "finetuning_task": null,
227
+ "forced_bos_token_id": null,
228
+ "forced_eos_token_id": null,
229
+ "id2label": {
230
+ "0": "LABEL_0",
231
+ "1": "LABEL_1"
232
+ },
233
+ "is_decoder": false,
234
+ "is_encoder_decoder": false,
235
+ "label2id": {
236
+ "LABEL_0": 0,
237
+ "LABEL_1": 1
238
+ },
239
+ "length_penalty": 1.0,
240
+ "llm_cfg": {
241
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/llm",
242
+ "add_cross_attention": false,
243
+ "architectures": [
244
+ "Qwen2ForCausalLM"
245
+ ],
246
+ "attention_dropout": 0.0,
247
+ "bad_words_ids": null,
248
+ "begin_suppress_tokens": null,
249
+ "bos_token_id": 151643,
250
+ "chunk_size_feed_forward": 0,
251
+ "cross_attention_hidden_size": null,
252
+ "decoder_start_token_id": null,
253
+ "diversity_penalty": 0.0,
254
+ "do_sample": false,
255
+ "early_stopping": false,
256
+ "encoder_no_repeat_ngram_size": 0,
257
+ "eos_token_id": 151645,
258
+ "exponential_decay_length_penalty": null,
259
+ "finetuning_task": null,
260
+ "forced_bos_token_id": null,
261
+ "forced_eos_token_id": null,
262
+ "hidden_act": "silu",
263
+ "hidden_size": 1536,
264
+ "id2label": {
265
+ "0": "LABEL_0",
266
+ "1": "LABEL_1"
267
+ },
268
+ "initializer_range": 0.02,
269
+ "intermediate_size": 8960,
270
+ "is_decoder": false,
271
+ "is_encoder_decoder": false,
272
+ "label2id": {
273
+ "LABEL_0": 0,
274
+ "LABEL_1": 1
275
+ },
276
+ "length_penalty": 1.0,
277
+ "max_length": 20,
278
+ "max_position_embeddings": 32768,
279
+ "max_window_layers": 28,
280
+ "min_length": 0,
281
+ "model_max_length": 4096,
282
+ "model_type": "qwen2",
283
+ "no_repeat_ngram_size": 0,
284
+ "num_attention_heads": 12,
285
+ "num_beam_groups": 1,
286
+ "num_beams": 1,
287
+ "num_hidden_layers": 28,
288
+ "num_key_value_heads": 2,
289
+ "num_return_sequences": 1,
290
+ "output_attentions": false,
291
+ "output_hidden_states": false,
292
+ "output_scores": false,
293
+ "pad_token_id": null,
294
+ "prefix": null,
295
+ "problem_type": null,
296
+ "pruned_heads": {},
297
+ "remove_invalid_values": false,
298
+ "repetition_penalty": 1.0,
299
+ "return_dict": true,
300
+ "return_dict_in_generate": false,
301
+ "rms_norm_eps": 1e-06,
302
+ "rope_theta": 1000000.0,
303
+ "sep_token_id": null,
304
+ "sliding_window": null,
305
+ "suppress_tokens": null,
306
+ "task_specific_params": null,
307
+ "temperature": 1.0,
308
+ "tf_legacy_loss": false,
309
+ "tie_encoder_decoder": false,
310
+ "tie_word_embeddings": true,
311
+ "tokenizer_class": null,
312
+ "tokenizer_model_max_length": 4096,
313
+ "tokenizer_padding_side": "right",
314
+ "top_k": 50,
315
+ "top_p": 1.0,
316
+ "torch_dtype": "bfloat16",
317
+ "torchscript": false,
318
+ "typical_p": 1.0,
319
+ "use_bfloat16": false,
320
+ "use_cache": true,
321
+ "use_sliding_window": false,
322
+ "vocab_size": 151648
323
+ },
324
+ "max_length": 20,
325
+ "min_length": 0,
326
+ "mm_projector_cfg": {
327
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/mm_projector",
328
+ "add_cross_attention": false,
329
+ "architectures": [
330
+ "MultimodalProjector"
331
+ ],
332
+ "bad_words_ids": null,
333
+ "begin_suppress_tokens": null,
334
+ "bos_token_id": null,
335
+ "chunk_size_feed_forward": 0,
336
+ "cross_attention_hidden_size": null,
337
+ "decoder_start_token_id": null,
338
+ "diversity_penalty": 0.0,
339
+ "do_sample": false,
340
+ "early_stopping": false,
341
+ "encoder_no_repeat_ngram_size": 0,
342
+ "eos_token_id": null,
343
+ "exponential_decay_length_penalty": null,
344
+ "finetuning_task": null,
345
+ "forced_bos_token_id": null,
346
+ "forced_eos_token_id": null,
347
+ "id2label": {
348
+ "0": "LABEL_0",
349
+ "1": "LABEL_1"
350
+ },
351
+ "is_decoder": false,
352
+ "is_encoder_decoder": false,
353
+ "label2id": {
354
+ "LABEL_0": 0,
355
+ "LABEL_1": 1
356
+ },
357
+ "length_penalty": 1.0,
358
+ "max_length": 20,
359
+ "min_length": 0,
360
+ "mm_projector_type": "mlp_downsample",
361
+ "model_type": "v2l_projector",
362
+ "no_repeat_ngram_size": 0,
363
+ "num_beam_groups": 1,
364
+ "num_beams": 1,
365
+ "num_return_sequences": 1,
366
+ "output_attentions": false,
367
+ "output_hidden_states": false,
368
+ "output_scores": false,
369
+ "pad_token_id": null,
370
+ "prefix": null,
371
+ "problem_type": null,
372
+ "pruned_heads": {},
373
+ "remove_invalid_values": false,
374
+ "repetition_penalty": 1.0,
375
+ "return_dict": true,
376
+ "return_dict_in_generate": false,
377
+ "sep_token_id": null,
378
+ "suppress_tokens": null,
379
+ "task_specific_params": null,
380
+ "temperature": 1.0,
381
+ "tf_legacy_loss": false,
382
+ "tie_encoder_decoder": false,
383
+ "tie_word_embeddings": true,
384
+ "tokenizer_class": null,
385
+ "top_k": 50,
386
+ "top_p": 1.0,
387
+ "torch_dtype": "bfloat16",
388
+ "torchscript": false,
389
+ "typical_p": 1.0,
390
+ "use_bfloat16": false
391
+ },
392
+ "model_type": "llava_llama",
393
+ "no_repeat_ngram_size": 0,
394
+ "num_beam_groups": 1,
395
+ "num_beams": 1,
396
+ "num_return_sequences": 1,
397
+ "output_attentions": false,
398
+ "output_hidden_states": false,
399
+ "output_scores": false,
400
+ "pad_token_id": null,
401
+ "prefix": null,
402
+ "problem_type": null,
403
+ "pruned_heads": {},
404
+ "remove_invalid_values": false,
405
+ "repetition_penalty": 1.0,
406
+ "return_dict": true,
407
+ "return_dict_in_generate": false,
408
+ "sep_token_id": null,
409
+ "suppress_tokens": null,
410
+ "task_specific_params": null,
411
+ "temperature": 1.0,
412
+ "tf_legacy_loss": false,
413
+ "tie_encoder_decoder": false,
414
+ "tie_word_embeddings": true,
415
+ "tokenizer_class": null,
416
+ "top_k": 50,
417
+ "top_p": 1.0,
418
+ "torch_dtype": "bfloat16",
419
+ "torchscript": false,
420
+ "typical_p": 1.0,
421
+ "use_bfloat16": false,
422
+ "vision_tower_cfg": {
423
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/vision_tower",
424
+ "add_cross_attention": false,
425
+ "architectures": [
426
+ "SiglipVisionModel"
427
+ ],
428
+ "attention_dropout": 0.0,
429
+ "bad_words_ids": null,
430
+ "begin_suppress_tokens": null,
431
+ "bos_token_id": null,
432
+ "chunk_size_feed_forward": 0,
433
+ "cross_attention_hidden_size": null,
434
+ "decoder_start_token_id": null,
435
+ "diversity_penalty": 0.0,
436
+ "do_sample": false,
437
+ "early_stopping": false,
438
+ "encoder_no_repeat_ngram_size": 0,
439
+ "eos_token_id": null,
440
+ "exponential_decay_length_penalty": null,
441
+ "finetuning_task": null,
442
+ "forced_bos_token_id": null,
443
+ "forced_eos_token_id": null,
444
+ "hidden_act": "gelu_pytorch_tanh",
445
+ "hidden_size": 1152,
446
+ "id2label": {
447
+ "0": "LABEL_0",
448
+ "1": "LABEL_1"
449
+ },
450
+ "image_size": 384,
451
+ "intermediate_size": 4304,
452
+ "is_decoder": false,
453
+ "is_encoder_decoder": false,
454
+ "label2id": {
455
+ "LABEL_0": 0,
456
+ "LABEL_1": 1
457
+ },
458
+ "layer_norm_eps": 1e-06,
459
+ "length_penalty": 1.0,
460
+ "max_length": 20,
461
+ "min_length": 0,
462
+ "model_type": "siglip_vision_model",
463
+ "no_repeat_ngram_size": 0,
464
+ "num_attention_heads": 16,
465
+ "num_beam_groups": 1,
466
+ "num_beams": 1,
467
+ "num_channels": 3,
468
+ "num_hidden_layers": 27,
469
+ "num_return_sequences": 1,
470
+ "output_attentions": false,
471
+ "output_hidden_states": false,
472
+ "output_scores": false,
473
+ "pad_token_id": null,
474
+ "patch_size": 14,
475
+ "prefix": null,
476
+ "problem_type": null,
477
+ "pruned_heads": {},
478
+ "remove_invalid_values": false,
479
+ "repetition_penalty": 1.0,
480
+ "return_dict": true,
481
+ "return_dict_in_generate": false,
482
+ "sep_token_id": null,
483
+ "suppress_tokens": null,
484
+ "task_specific_params": null,
485
+ "temperature": 1.0,
486
+ "tf_legacy_loss": false,
487
+ "tie_encoder_decoder": false,
488
+ "tie_word_embeddings": true,
489
+ "tokenizer_class": null,
490
+ "top_k": 50,
491
+ "top_p": 1.0,
492
+ "torch_dtype": "bfloat16",
493
+ "torchscript": false,
494
+ "typical_p": 1.0,
495
+ "use_bfloat16": false
496
+ }
497
+ },
498
+ "traj_decoder_type": "transformer_split_action_v2",
499
+ "transformers_version": "4.44.2",
500
+ "tune_language_model": true,
501
+ "tune_mm_projector": true,
502
+ "tune_vision_tower": true,
503
+ "use_movement_mask": false,
504
+ "use_proprio": true,
505
+ "vision_resolution": -1,
506
+ "vision_tower_cfg": {
507
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/vision_tower",
508
+ "add_cross_attention": false,
509
+ "architectures": [
510
+ "SiglipVisionModel"
511
+ ],
512
+ "attention_dropout": 0.0,
513
+ "bad_words_ids": null,
514
+ "begin_suppress_tokens": null,
515
+ "bos_token_id": null,
516
+ "chunk_size_feed_forward": 0,
517
+ "cross_attention_hidden_size": null,
518
+ "decoder_start_token_id": null,
519
+ "diversity_penalty": 0.0,
520
+ "do_sample": false,
521
+ "early_stopping": false,
522
+ "encoder_no_repeat_ngram_size": 0,
523
+ "eos_token_id": null,
524
+ "exponential_decay_length_penalty": null,
525
+ "finetuning_task": null,
526
+ "forced_bos_token_id": null,
527
+ "forced_eos_token_id": null,
528
+ "hidden_act": "gelu_pytorch_tanh",
529
+ "hidden_size": 1152,
530
+ "id2label": {
531
+ "0": "LABEL_0",
532
+ "1": "LABEL_1"
533
+ },
534
+ "image_size": 384,
535
+ "intermediate_size": 4304,
536
+ "is_decoder": false,
537
+ "is_encoder_decoder": false,
538
+ "label2id": {
539
+ "LABEL_0": 0,
540
+ "LABEL_1": 1
541
+ },
542
+ "layer_norm_eps": 1e-06,
543
+ "length_penalty": 1.0,
544
+ "max_length": 20,
545
+ "min_length": 0,
546
+ "model_type": "siglip_vision_model",
547
+ "no_repeat_ngram_size": 0,
548
+ "num_attention_heads": 16,
549
+ "num_beam_groups": 1,
550
+ "num_beams": 1,
551
+ "num_channels": 3,
552
+ "num_hidden_layers": 27,
553
+ "num_return_sequences": 1,
554
+ "output_attentions": false,
555
+ "output_hidden_states": false,
556
+ "output_scores": false,
557
+ "pad_token_id": null,
558
+ "patch_size": 14,
559
+ "prefix": null,
560
+ "problem_type": null,
561
+ "pruned_heads": {},
562
+ "remove_invalid_values": false,
563
+ "repetition_penalty": 1.0,
564
+ "return_dict": true,
565
+ "return_dict_in_generate": false,
566
+ "sep_token_id": null,
567
+ "suppress_tokens": null,
568
+ "task_specific_params": null,
569
+ "temperature": 1.0,
570
+ "tf_legacy_loss": false,
571
+ "tie_encoder_decoder": false,
572
+ "tie_word_embeddings": true,
573
+ "tokenizer_class": null,
574
+ "top_k": 50,
575
+ "top_p": 1.0,
576
+ "torch_dtype": "bfloat16",
577
+ "torchscript": false,
578
+ "typical_p": 1.0,
579
+ "use_bfloat16": false
580
+ }
581
+ }
ego_vla_checkpoint/ckpt-6720/llm/added_tokens.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "<|endoftext|>": 151643,
3
+ "<|im_end|>": 151645,
4
+ "<|im_start|>": 151644,
5
+ "[BOS]": 151646,
6
+ "[PAD]": 151647
7
+ }
ego_vla_checkpoint/ckpt-6720/llm/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/llm",
3
+ "architectures": [
4
+ "Qwen2ForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "eos_token_id": 151645,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 1536,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 8960,
13
+ "max_position_embeddings": 32768,
14
+ "max_window_layers": 28,
15
+ "model_max_length": 4096,
16
+ "model_type": "qwen2",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 28,
19
+ "num_key_value_heads": 2,
20
+ "rms_norm_eps": 1e-06,
21
+ "rope_theta": 1000000.0,
22
+ "sliding_window": null,
23
+ "tie_word_embeddings": true,
24
+ "tokenizer_model_max_length": 4096,
25
+ "tokenizer_padding_side": "right",
26
+ "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.44.2",
28
+ "use_cache": true,
29
+ "use_sliding_window": false,
30
+ "vocab_size": 151648
31
+ }
ego_vla_checkpoint/ckpt-6720/llm/generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 151645,
6
+ 151643
7
+ ],
8
+ "pad_token_id": 151643,
9
+ "repetition_penalty": 1.1,
10
+ "temperature": 0.7,
11
+ "top_k": 20,
12
+ "top_p": 0.8,
13
+ "transformers_version": "4.44.2"
14
+ }
ego_vla_checkpoint/ckpt-6720/llm/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
ego_vla_checkpoint/ckpt-6720/llm/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f3dcf899d2a35a1fb229af0cbbe5fd27843938a4968679a9e5b0e55367802cf
3
+ size 3086582408
ego_vla_checkpoint/ckpt-6720/llm/special_tokens_map.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>"
5
+ ],
6
+ "bos_token": {
7
+ "content": "[BOS]",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false
12
+ },
13
+ "eos_token": {
14
+ "content": "<|im_end|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false
19
+ },
20
+ "pad_token": {
21
+ "content": "[PAD]",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ }
27
+ }
ego_vla_checkpoint/ckpt-6720/llm/tokenizer_config.json ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "151643": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "151644": {
13
+ "content": "<|im_start|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "151645": {
21
+ "content": "<|im_end|>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "151646": {
29
+ "content": "[BOS]",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "151647": {
37
+ "content": "[PAD]",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "additional_special_tokens": [
46
+ "<|im_start|>",
47
+ "<|im_end|>"
48
+ ],
49
+ "bos_token": "[BOS]",
50
+ "chat_template": "{% if messages[0]['role'] != 'system' %}{{ '<|im_start|>system\\nYou are a helpful assistant<|im_end|>\\n' }}{% endif %}{% for message in messages %}{% if message['content'] is not none %}{{ '<|im_start|>' + message['role'] + '\\n' + message['content'] + '<|im_end|>' + '\\n' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\\n' }}{% endif %}",
51
+ "clean_up_tokenization_spaces": false,
52
+ "eos_token": "<|im_end|>",
53
+ "errors": "replace",
54
+ "legacy": false,
55
+ "model_max_length": 4096,
56
+ "pad_token": "[PAD]",
57
+ "padding_side": "right",
58
+ "split_special_tokens": false,
59
+ "tokenizer_class": "Qwen2Tokenizer",
60
+ "unk_token": null
61
+ }
ego_vla_checkpoint/ckpt-6720/llm/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
ego_vla_checkpoint/ckpt-6720/mm_projector/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/mm_projector",
3
+ "architectures": [
4
+ "MultimodalProjector"
5
+ ],
6
+ "mm_projector_type": "mlp_downsample",
7
+ "model_type": "v2l_projector",
8
+ "torch_dtype": "bfloat16",
9
+ "transformers_version": "4.44.2"
10
+ }
ego_vla_checkpoint/ckpt-6720/mm_projector/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:498ea96407600c15553fceebddaf0d4aa3728e1af89674ed85deeaab82349591
3
+ size 18899472
ego_vla_checkpoint/ckpt-6720/traj_decoder/config.json ADDED
@@ -0,0 +1,581 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/traj_decoder",
3
+ "action_output_dim": 48,
4
+ "action_output_ee_2d_dim": 0,
5
+ "action_output_ee_dim": 6,
6
+ "action_output_ee_rot_dim": 12,
7
+ "action_output_hand_dim": 30,
8
+ "action_token_idx": 31733,
9
+ "architectures": [
10
+ "TrajDecoder"
11
+ ],
12
+ "chat_template": null,
13
+ "cvae_hidden_size": 256,
14
+ "cvae_latent_dim": 256,
15
+ "drop_path_rate": 0.0,
16
+ "ee_2d_loss_coeff": 0.0,
17
+ "ee_loss_coeff": 20.0,
18
+ "ee_movement_loss_weight": 1,
19
+ "ee_rot_loss_coeff": 5.0,
20
+ "ee_rot_representation": "rot6d",
21
+ "fps": 0.0,
22
+ "hand_kp_loss_coeff": 0.0,
23
+ "hand_loss_coeff": 5.0,
24
+ "hand_loss_dim": 6,
25
+ "hidden_size": 1536,
26
+ "image_aspect_ratio": "resize",
27
+ "input_placeholder_end_token_idx": 151195,
28
+ "input_placeholder_start_token_idx": 151375,
29
+ "input_placeholder_token_idx": 151375,
30
+ "interpolate_mode": "linear",
31
+ "invalid_token_idx": 151376,
32
+ "invalid_token_weight": 0.1,
33
+ "kl_weight": 0.001,
34
+ "llm_cfg": {
35
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/llm",
36
+ "add_cross_attention": false,
37
+ "architectures": [
38
+ "Qwen2ForCausalLM"
39
+ ],
40
+ "attention_dropout": 0.0,
41
+ "bad_words_ids": null,
42
+ "begin_suppress_tokens": null,
43
+ "bos_token_id": 151643,
44
+ "chunk_size_feed_forward": 0,
45
+ "cross_attention_hidden_size": null,
46
+ "decoder_start_token_id": null,
47
+ "diversity_penalty": 0.0,
48
+ "do_sample": false,
49
+ "early_stopping": false,
50
+ "encoder_no_repeat_ngram_size": 0,
51
+ "eos_token_id": 151645,
52
+ "exponential_decay_length_penalty": null,
53
+ "finetuning_task": null,
54
+ "forced_bos_token_id": null,
55
+ "forced_eos_token_id": null,
56
+ "hidden_act": "silu",
57
+ "hidden_size": 1536,
58
+ "id2label": {
59
+ "0": "LABEL_0",
60
+ "1": "LABEL_1"
61
+ },
62
+ "initializer_range": 0.02,
63
+ "intermediate_size": 8960,
64
+ "is_decoder": false,
65
+ "is_encoder_decoder": false,
66
+ "label2id": {
67
+ "LABEL_0": 0,
68
+ "LABEL_1": 1
69
+ },
70
+ "length_penalty": 1.0,
71
+ "max_length": 20,
72
+ "max_position_embeddings": 32768,
73
+ "max_window_layers": 28,
74
+ "min_length": 0,
75
+ "model_max_length": 4096,
76
+ "model_type": "qwen2",
77
+ "no_repeat_ngram_size": 0,
78
+ "num_attention_heads": 12,
79
+ "num_beam_groups": 1,
80
+ "num_beams": 1,
81
+ "num_hidden_layers": 28,
82
+ "num_key_value_heads": 2,
83
+ "num_return_sequences": 1,
84
+ "output_attentions": false,
85
+ "output_hidden_states": false,
86
+ "output_scores": false,
87
+ "pad_token_id": null,
88
+ "prefix": null,
89
+ "problem_type": null,
90
+ "pruned_heads": {},
91
+ "remove_invalid_values": false,
92
+ "repetition_penalty": 1.0,
93
+ "return_dict": true,
94
+ "return_dict_in_generate": false,
95
+ "rms_norm_eps": 1e-06,
96
+ "rope_theta": 1000000.0,
97
+ "sep_token_id": null,
98
+ "sliding_window": null,
99
+ "suppress_tokens": null,
100
+ "task_specific_params": null,
101
+ "temperature": 1.0,
102
+ "tf_legacy_loss": false,
103
+ "tie_encoder_decoder": false,
104
+ "tie_word_embeddings": true,
105
+ "tokenizer_class": null,
106
+ "tokenizer_model_max_length": 4096,
107
+ "tokenizer_padding_side": "right",
108
+ "top_k": 50,
109
+ "top_p": 1.0,
110
+ "torch_dtype": "bfloat16",
111
+ "torchscript": false,
112
+ "typical_p": 1.0,
113
+ "use_bfloat16": false,
114
+ "use_cache": true,
115
+ "use_sliding_window": false,
116
+ "vocab_size": 151648
117
+ },
118
+ "loss_use_l1": true,
119
+ "merge_hand": true,
120
+ "mm_hidden_size": 1152,
121
+ "mm_projector_cfg": {
122
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/mm_projector",
123
+ "add_cross_attention": false,
124
+ "architectures": [
125
+ "MultimodalProjector"
126
+ ],
127
+ "bad_words_ids": null,
128
+ "begin_suppress_tokens": null,
129
+ "bos_token_id": null,
130
+ "chunk_size_feed_forward": 0,
131
+ "cross_attention_hidden_size": null,
132
+ "decoder_start_token_id": null,
133
+ "diversity_penalty": 0.0,
134
+ "do_sample": false,
135
+ "early_stopping": false,
136
+ "encoder_no_repeat_ngram_size": 0,
137
+ "eos_token_id": null,
138
+ "exponential_decay_length_penalty": null,
139
+ "finetuning_task": null,
140
+ "forced_bos_token_id": null,
141
+ "forced_eos_token_id": null,
142
+ "id2label": {
143
+ "0": "LABEL_0",
144
+ "1": "LABEL_1"
145
+ },
146
+ "is_decoder": false,
147
+ "is_encoder_decoder": false,
148
+ "label2id": {
149
+ "LABEL_0": 0,
150
+ "LABEL_1": 1
151
+ },
152
+ "length_penalty": 1.0,
153
+ "max_length": 20,
154
+ "min_length": 0,
155
+ "mm_projector_type": "mlp_downsample",
156
+ "model_type": "v2l_projector",
157
+ "no_repeat_ngram_size": 0,
158
+ "num_beam_groups": 1,
159
+ "num_beams": 1,
160
+ "num_return_sequences": 1,
161
+ "output_attentions": false,
162
+ "output_hidden_states": false,
163
+ "output_scores": false,
164
+ "pad_token_id": null,
165
+ "prefix": null,
166
+ "problem_type": null,
167
+ "pruned_heads": {},
168
+ "remove_invalid_values": false,
169
+ "repetition_penalty": 1.0,
170
+ "return_dict": true,
171
+ "return_dict_in_generate": false,
172
+ "sep_token_id": null,
173
+ "suppress_tokens": null,
174
+ "task_specific_params": null,
175
+ "temperature": 1.0,
176
+ "tf_legacy_loss": false,
177
+ "tie_encoder_decoder": false,
178
+ "tie_word_embeddings": true,
179
+ "tokenizer_class": null,
180
+ "top_k": 50,
181
+ "top_p": 1.0,
182
+ "torch_dtype": "bfloat16",
183
+ "torchscript": false,
184
+ "typical_p": 1.0,
185
+ "use_bfloat16": false
186
+ },
187
+ "mm_projector_lr": null,
188
+ "mm_use_im_patch_token": true,
189
+ "mm_use_im_start_end": false,
190
+ "mm_vision_select_feature": "cls_patch",
191
+ "mm_vision_select_layer": -2,
192
+ "model_dtype": "torch.bfloat16",
193
+ "model_type": "llava_llama",
194
+ "next_token_loss_coeff": 0.0,
195
+ "num_time_tokens": 0,
196
+ "num_video_frames": 8,
197
+ "proprio_size": 16,
198
+ "resume_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4",
199
+ "s2": false,
200
+ "s2_max_split_size": 336,
201
+ "s2_scales": "336,672,1008",
202
+ "sep_proprio": true,
203
+ "sep_query_token": true,
204
+ "soft_ce_std": 1.0,
205
+ "time_token_format": "<t{t}>",
206
+ "time_token_ids": [],
207
+ "torch_dtype": "bfloat16",
208
+ "traj_decoder_cfg": {
209
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/traj_decoder",
210
+ "add_cross_attention": false,
211
+ "architectures": [
212
+ "TrajDecoder"
213
+ ],
214
+ "bad_words_ids": null,
215
+ "begin_suppress_tokens": null,
216
+ "bos_token_id": null,
217
+ "chunk_size_feed_forward": 0,
218
+ "cross_attention_hidden_size": null,
219
+ "decoder_start_token_id": null,
220
+ "diversity_penalty": 0.0,
221
+ "do_sample": false,
222
+ "early_stopping": false,
223
+ "encoder_no_repeat_ngram_size": 0,
224
+ "eos_token_id": null,
225
+ "exponential_decay_length_penalty": null,
226
+ "finetuning_task": null,
227
+ "forced_bos_token_id": null,
228
+ "forced_eos_token_id": null,
229
+ "id2label": {
230
+ "0": "LABEL_0",
231
+ "1": "LABEL_1"
232
+ },
233
+ "is_decoder": false,
234
+ "is_encoder_decoder": false,
235
+ "label2id": {
236
+ "LABEL_0": 0,
237
+ "LABEL_1": 1
238
+ },
239
+ "length_penalty": 1.0,
240
+ "llm_cfg": {
241
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/llm",
242
+ "add_cross_attention": false,
243
+ "architectures": [
244
+ "Qwen2ForCausalLM"
245
+ ],
246
+ "attention_dropout": 0.0,
247
+ "bad_words_ids": null,
248
+ "begin_suppress_tokens": null,
249
+ "bos_token_id": 151643,
250
+ "chunk_size_feed_forward": 0,
251
+ "cross_attention_hidden_size": null,
252
+ "decoder_start_token_id": null,
253
+ "diversity_penalty": 0.0,
254
+ "do_sample": false,
255
+ "early_stopping": false,
256
+ "encoder_no_repeat_ngram_size": 0,
257
+ "eos_token_id": 151645,
258
+ "exponential_decay_length_penalty": null,
259
+ "finetuning_task": null,
260
+ "forced_bos_token_id": null,
261
+ "forced_eos_token_id": null,
262
+ "hidden_act": "silu",
263
+ "hidden_size": 1536,
264
+ "id2label": {
265
+ "0": "LABEL_0",
266
+ "1": "LABEL_1"
267
+ },
268
+ "initializer_range": 0.02,
269
+ "intermediate_size": 8960,
270
+ "is_decoder": false,
271
+ "is_encoder_decoder": false,
272
+ "label2id": {
273
+ "LABEL_0": 0,
274
+ "LABEL_1": 1
275
+ },
276
+ "length_penalty": 1.0,
277
+ "max_length": 20,
278
+ "max_position_embeddings": 32768,
279
+ "max_window_layers": 28,
280
+ "min_length": 0,
281
+ "model_max_length": 4096,
282
+ "model_type": "qwen2",
283
+ "no_repeat_ngram_size": 0,
284
+ "num_attention_heads": 12,
285
+ "num_beam_groups": 1,
286
+ "num_beams": 1,
287
+ "num_hidden_layers": 28,
288
+ "num_key_value_heads": 2,
289
+ "num_return_sequences": 1,
290
+ "output_attentions": false,
291
+ "output_hidden_states": false,
292
+ "output_scores": false,
293
+ "pad_token_id": null,
294
+ "prefix": null,
295
+ "problem_type": null,
296
+ "pruned_heads": {},
297
+ "remove_invalid_values": false,
298
+ "repetition_penalty": 1.0,
299
+ "return_dict": true,
300
+ "return_dict_in_generate": false,
301
+ "rms_norm_eps": 1e-06,
302
+ "rope_theta": 1000000.0,
303
+ "sep_token_id": null,
304
+ "sliding_window": null,
305
+ "suppress_tokens": null,
306
+ "task_specific_params": null,
307
+ "temperature": 1.0,
308
+ "tf_legacy_loss": false,
309
+ "tie_encoder_decoder": false,
310
+ "tie_word_embeddings": true,
311
+ "tokenizer_class": null,
312
+ "tokenizer_model_max_length": 4096,
313
+ "tokenizer_padding_side": "right",
314
+ "top_k": 50,
315
+ "top_p": 1.0,
316
+ "torch_dtype": "bfloat16",
317
+ "torchscript": false,
318
+ "typical_p": 1.0,
319
+ "use_bfloat16": false,
320
+ "use_cache": true,
321
+ "use_sliding_window": false,
322
+ "vocab_size": 151648
323
+ },
324
+ "max_length": 20,
325
+ "min_length": 0,
326
+ "mm_projector_cfg": {
327
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/mm_projector",
328
+ "add_cross_attention": false,
329
+ "architectures": [
330
+ "MultimodalProjector"
331
+ ],
332
+ "bad_words_ids": null,
333
+ "begin_suppress_tokens": null,
334
+ "bos_token_id": null,
335
+ "chunk_size_feed_forward": 0,
336
+ "cross_attention_hidden_size": null,
337
+ "decoder_start_token_id": null,
338
+ "diversity_penalty": 0.0,
339
+ "do_sample": false,
340
+ "early_stopping": false,
341
+ "encoder_no_repeat_ngram_size": 0,
342
+ "eos_token_id": null,
343
+ "exponential_decay_length_penalty": null,
344
+ "finetuning_task": null,
345
+ "forced_bos_token_id": null,
346
+ "forced_eos_token_id": null,
347
+ "id2label": {
348
+ "0": "LABEL_0",
349
+ "1": "LABEL_1"
350
+ },
351
+ "is_decoder": false,
352
+ "is_encoder_decoder": false,
353
+ "label2id": {
354
+ "LABEL_0": 0,
355
+ "LABEL_1": 1
356
+ },
357
+ "length_penalty": 1.0,
358
+ "max_length": 20,
359
+ "min_length": 0,
360
+ "mm_projector_type": "mlp_downsample",
361
+ "model_type": "v2l_projector",
362
+ "no_repeat_ngram_size": 0,
363
+ "num_beam_groups": 1,
364
+ "num_beams": 1,
365
+ "num_return_sequences": 1,
366
+ "output_attentions": false,
367
+ "output_hidden_states": false,
368
+ "output_scores": false,
369
+ "pad_token_id": null,
370
+ "prefix": null,
371
+ "problem_type": null,
372
+ "pruned_heads": {},
373
+ "remove_invalid_values": false,
374
+ "repetition_penalty": 1.0,
375
+ "return_dict": true,
376
+ "return_dict_in_generate": false,
377
+ "sep_token_id": null,
378
+ "suppress_tokens": null,
379
+ "task_specific_params": null,
380
+ "temperature": 1.0,
381
+ "tf_legacy_loss": false,
382
+ "tie_encoder_decoder": false,
383
+ "tie_word_embeddings": true,
384
+ "tokenizer_class": null,
385
+ "top_k": 50,
386
+ "top_p": 1.0,
387
+ "torch_dtype": "bfloat16",
388
+ "torchscript": false,
389
+ "typical_p": 1.0,
390
+ "use_bfloat16": false
391
+ },
392
+ "model_type": "llava_llama",
393
+ "no_repeat_ngram_size": 0,
394
+ "num_beam_groups": 1,
395
+ "num_beams": 1,
396
+ "num_return_sequences": 1,
397
+ "output_attentions": false,
398
+ "output_hidden_states": false,
399
+ "output_scores": false,
400
+ "pad_token_id": null,
401
+ "prefix": null,
402
+ "problem_type": null,
403
+ "pruned_heads": {},
404
+ "remove_invalid_values": false,
405
+ "repetition_penalty": 1.0,
406
+ "return_dict": true,
407
+ "return_dict_in_generate": false,
408
+ "sep_token_id": null,
409
+ "suppress_tokens": null,
410
+ "task_specific_params": null,
411
+ "temperature": 1.0,
412
+ "tf_legacy_loss": false,
413
+ "tie_encoder_decoder": false,
414
+ "tie_word_embeddings": true,
415
+ "tokenizer_class": null,
416
+ "top_k": 50,
417
+ "top_p": 1.0,
418
+ "torch_dtype": "bfloat16",
419
+ "torchscript": false,
420
+ "typical_p": 1.0,
421
+ "use_bfloat16": false,
422
+ "vision_tower_cfg": {
423
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/vision_tower",
424
+ "add_cross_attention": false,
425
+ "architectures": [
426
+ "SiglipVisionModel"
427
+ ],
428
+ "attention_dropout": 0.0,
429
+ "bad_words_ids": null,
430
+ "begin_suppress_tokens": null,
431
+ "bos_token_id": null,
432
+ "chunk_size_feed_forward": 0,
433
+ "cross_attention_hidden_size": null,
434
+ "decoder_start_token_id": null,
435
+ "diversity_penalty": 0.0,
436
+ "do_sample": false,
437
+ "early_stopping": false,
438
+ "encoder_no_repeat_ngram_size": 0,
439
+ "eos_token_id": null,
440
+ "exponential_decay_length_penalty": null,
441
+ "finetuning_task": null,
442
+ "forced_bos_token_id": null,
443
+ "forced_eos_token_id": null,
444
+ "hidden_act": "gelu_pytorch_tanh",
445
+ "hidden_size": 1152,
446
+ "id2label": {
447
+ "0": "LABEL_0",
448
+ "1": "LABEL_1"
449
+ },
450
+ "image_size": 384,
451
+ "intermediate_size": 4304,
452
+ "is_decoder": false,
453
+ "is_encoder_decoder": false,
454
+ "label2id": {
455
+ "LABEL_0": 0,
456
+ "LABEL_1": 1
457
+ },
458
+ "layer_norm_eps": 1e-06,
459
+ "length_penalty": 1.0,
460
+ "max_length": 20,
461
+ "min_length": 0,
462
+ "model_type": "siglip_vision_model",
463
+ "no_repeat_ngram_size": 0,
464
+ "num_attention_heads": 16,
465
+ "num_beam_groups": 1,
466
+ "num_beams": 1,
467
+ "num_channels": 3,
468
+ "num_hidden_layers": 27,
469
+ "num_return_sequences": 1,
470
+ "output_attentions": false,
471
+ "output_hidden_states": false,
472
+ "output_scores": false,
473
+ "pad_token_id": null,
474
+ "patch_size": 14,
475
+ "prefix": null,
476
+ "problem_type": null,
477
+ "pruned_heads": {},
478
+ "remove_invalid_values": false,
479
+ "repetition_penalty": 1.0,
480
+ "return_dict": true,
481
+ "return_dict_in_generate": false,
482
+ "sep_token_id": null,
483
+ "suppress_tokens": null,
484
+ "task_specific_params": null,
485
+ "temperature": 1.0,
486
+ "tf_legacy_loss": false,
487
+ "tie_encoder_decoder": false,
488
+ "tie_word_embeddings": true,
489
+ "tokenizer_class": null,
490
+ "top_k": 50,
491
+ "top_p": 1.0,
492
+ "torch_dtype": "bfloat16",
493
+ "torchscript": false,
494
+ "typical_p": 1.0,
495
+ "use_bfloat16": false
496
+ }
497
+ },
498
+ "traj_decoder_type": "transformer_split_action_v2",
499
+ "transformers_version": "4.44.2",
500
+ "tune_language_model": true,
501
+ "tune_mm_projector": true,
502
+ "tune_vision_tower": true,
503
+ "use_movement_mask": false,
504
+ "use_proprio": true,
505
+ "vision_resolution": -1,
506
+ "vision_tower_cfg": {
507
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/vision_tower",
508
+ "add_cross_attention": false,
509
+ "architectures": [
510
+ "SiglipVisionModel"
511
+ ],
512
+ "attention_dropout": 0.0,
513
+ "bad_words_ids": null,
514
+ "begin_suppress_tokens": null,
515
+ "bos_token_id": null,
516
+ "chunk_size_feed_forward": 0,
517
+ "cross_attention_hidden_size": null,
518
+ "decoder_start_token_id": null,
519
+ "diversity_penalty": 0.0,
520
+ "do_sample": false,
521
+ "early_stopping": false,
522
+ "encoder_no_repeat_ngram_size": 0,
523
+ "eos_token_id": null,
524
+ "exponential_decay_length_penalty": null,
525
+ "finetuning_task": null,
526
+ "forced_bos_token_id": null,
527
+ "forced_eos_token_id": null,
528
+ "hidden_act": "gelu_pytorch_tanh",
529
+ "hidden_size": 1152,
530
+ "id2label": {
531
+ "0": "LABEL_0",
532
+ "1": "LABEL_1"
533
+ },
534
+ "image_size": 384,
535
+ "intermediate_size": 4304,
536
+ "is_decoder": false,
537
+ "is_encoder_decoder": false,
538
+ "label2id": {
539
+ "LABEL_0": 0,
540
+ "LABEL_1": 1
541
+ },
542
+ "layer_norm_eps": 1e-06,
543
+ "length_penalty": 1.0,
544
+ "max_length": 20,
545
+ "min_length": 0,
546
+ "model_type": "siglip_vision_model",
547
+ "no_repeat_ngram_size": 0,
548
+ "num_attention_heads": 16,
549
+ "num_beam_groups": 1,
550
+ "num_beams": 1,
551
+ "num_channels": 3,
552
+ "num_hidden_layers": 27,
553
+ "num_return_sequences": 1,
554
+ "output_attentions": false,
555
+ "output_hidden_states": false,
556
+ "output_scores": false,
557
+ "pad_token_id": null,
558
+ "patch_size": 14,
559
+ "prefix": null,
560
+ "problem_type": null,
561
+ "pruned_heads": {},
562
+ "remove_invalid_values": false,
563
+ "repetition_penalty": 1.0,
564
+ "return_dict": true,
565
+ "return_dict_in_generate": false,
566
+ "sep_token_id": null,
567
+ "suppress_tokens": null,
568
+ "task_specific_params": null,
569
+ "temperature": 1.0,
570
+ "tf_legacy_loss": false,
571
+ "tie_encoder_decoder": false,
572
+ "tie_word_embeddings": true,
573
+ "tokenizer_class": null,
574
+ "top_k": 50,
575
+ "top_p": 1.0,
576
+ "torch_dtype": "bfloat16",
577
+ "torchscript": false,
578
+ "typical_p": 1.0,
579
+ "use_bfloat16": false
580
+ }
581
+ }
ego_vla_checkpoint/ckpt-6720/traj_decoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac2dee51fa74138f70a54b6bb81cb1a7f27ee64e464c69b91d5c7fb260cf7ab2
3
+ size 217554240
ego_vla_checkpoint/ckpt-6720/vision_tower/config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pretrain_transv2update2_diffdata/otv-fixed-set-aug-room1table1-fixorder-30hz-hoi4dhot3dholotaco-transv2-100e-h5-3d20-rot5-lr2e-6-const-h5p30f1skip6-manoloss6l1-kp0-b16-4/vision_tower",
3
+ "architectures": [
4
+ "SiglipVisionModel"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "hidden_act": "gelu_pytorch_tanh",
8
+ "hidden_size": 1152,
9
+ "image_size": 384,
10
+ "intermediate_size": 4304,
11
+ "layer_norm_eps": 1e-06,
12
+ "model_type": "siglip_vision_model",
13
+ "num_attention_heads": 16,
14
+ "num_channels": 3,
15
+ "num_hidden_layers": 27,
16
+ "patch_size": 14,
17
+ "torch_dtype": "bfloat16",
18
+ "transformers_version": "4.44.2"
19
+ }
ego_vla_checkpoint/ckpt-6720/vision_tower/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de8670a93e897dbfcba37c8b3e155be019eeee72eff077cdf79b1c5b016f17cc
3
+ size 856506120
ego_vla_checkpoint/ckpt-6720/vision_tower/preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "SiglipImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "processor_class": "SiglipProcessor",
18
+ "resample": 3,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "height": 384,
22
+ "width": 384
23
+ }
24
+ }