hetolin commited on
Commit
16d0df3
·
verified ·
1 Parent(s): bd958cd

Upload PoseVLA checkpoint

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "pi0_ours",
3
+ "n_obs_steps": 1,
4
+ "normalization_mapping": {
5
+ "VISUAL": "IDENTITY",
6
+ "STATE": "IDENTITY",
7
+ "ACTION": "IDENTITY"
8
+ },
9
+ "input_features": {},
10
+ "output_features": {},
11
+ "chunk_size": 50,
12
+ "n_action_steps": 50,
13
+ "max_state_dim": 32,
14
+ "max_action_dim": 32,
15
+ "resize_imgs_with_padding": [
16
+ 224,
17
+ 224
18
+ ],
19
+ "empty_cameras": 0,
20
+ "tokenizer_max_length": 512,
21
+ "tokenizer_model_path": "google/paligemma-3b-pt-224",
22
+ "proj_width": 1024,
23
+ "num_steps": 10,
24
+ "use_cache": true,
25
+ "attention_implementation": "eager",
26
+ "pi05": false,
27
+ "is_knowledge_insulation": false,
28
+ "add_extra_token": true,
29
+ "add_image_token": true,
30
+ "add_prior": true,
31
+ "freeze_vision_encoder": false,
32
+ "train_expert_only": false,
33
+ "train_state_proj": true,
34
+ "optimizer_lr": 5e-05,
35
+ "optimizer_betas": [
36
+ 0.9,
37
+ 0.95
38
+ ],
39
+ "optimizer_eps": 1e-08,
40
+ "optimizer_weight_decay": 1e-10,
41
+ "scheduler_warmup_steps": 16000,
42
+ "scheduler_decay_steps": 1280000,
43
+ "scheduler_decay_lr": 5e-06,
44
+ "vis_attn": true,
45
+ "skip_init_weights": false,
46
+ "device": "cpu"
47
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:206742d74694c645f57d9dca4859de17c03f789b2d85c5afed4b4ea792ef09bd
3
+ size 6533190976
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<image>"
4
+ ],
5
+ "bos_token": {
6
+ "content": "<bos>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "eos_token": {
13
+ "content": "<eos>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "pad_token": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:096a73268dbf1a04677304baf37ee8f3c04fcaac74870fc7161147b40802f8b1
3
+ size 35828586
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff