Hugging Face's logo

yinchenghust
/
openpi_base

Safetensors
pi0
Model card Files
xet
Community
yinchenghust commited on
Commit
a5c799a
·
verified ·
1 Parent(s): 14a353b

Upload PI0Policy

Browse files
Files changed (5) hide show
  1. config.json +150 -0
  2. model-00001-of-00003.safetensors +3 -0
  3. model-00002-of-00003.safetensors +3 -0
  4. model-00003-of-00003.safetensors +3 -0
  5. model.safetensors.index.json +0 -0
config.json ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "physical-intelligence/pi0_base",
3
+ "action_delta_indices": [
4
+ 0,
5
+ 1,
6
+ 2,
7
+ 3,
8
+ 4,
9
+ 5,
10
+ 6,
11
+ 7,
12
+ 8,
13
+ 9,
14
+ 10,
15
+ 11,
16
+ 12,
17
+ 13,
18
+ 14,
19
+ 15,
20
+ 16,
21
+ 17,
22
+ 18,
23
+ 19,
24
+ 20,
25
+ 21,
26
+ 22,
27
+ 23,
28
+ 24,
29
+ 25,
30
+ 26,
31
+ 27,
32
+ 28,
33
+ 29,
34
+ 30,
35
+ 31,
36
+ 32,
37
+ 33,
38
+ 34,
39
+ 35,
40
+ 36,
41
+ 37,
42
+ 38,
43
+ 39,
44
+ 40,
45
+ 41,
46
+ 42,
47
+ 43,
48
+ 44,
49
+ 45,
50
+ 46,
51
+ 47,
52
+ 48,
53
+ 49
54
+ ],
55
+ "action_feature": {
56
+ "shape": [
57
+ 7
58
+ ],
59
+ "type": "ACTION"
60
+ },
61
+ "architectures": [
62
+ "PI0Policy"
63
+ ],
64
+ "attention_implementation": "eager",
65
+ "attn_implementation": null,
66
+ "denoise_num_steps": 10,
67
+ "empty_cameras": 0,
68
+ "env_state_feature": null,
69
+ "freeze_vision_encoder": true,
70
+ "hidden_size": 1024,
71
+ "image_features": {
72
+ "observation.images.image": {
73
+ "shape": [
74
+ 3,
75
+ 256,
76
+ 256
77
+ ],
78
+ "type": "VISUAL"
79
+ },
80
+ "observation.images.wrist_image": {
81
+ "shape": [
82
+ 3,
83
+ 256,
84
+ 256
85
+ ],
86
+ "type": "VISUAL"
87
+ }
88
+ },
89
+ "input_features": {
90
+ "observation.images.image": {
91
+ "shape": [
92
+ 3,
93
+ 256,
94
+ 256
95
+ ],
96
+ "type": "VISUAL"
97
+ },
98
+ "observation.images.wrist_image": {
99
+ "shape": [
100
+ 3,
101
+ 256,
102
+ 256
103
+ ],
104
+ "type": "VISUAL"
105
+ },
106
+ "observation.state": {
107
+ "shape": [
108
+ 8
109
+ ],
110
+ "type": "STATE"
111
+ }
112
+ },
113
+ "max_action_dim": 32,
114
+ "max_state_dim": 32,
115
+ "model_type": "pi0",
116
+ "n_action_steps": 50,
117
+ "n_obs_steps": 1,
118
+ "normalization_mapping": {
119
+ "ACTION": "MEAN_STD",
120
+ "STATE": "MEAN_STD",
121
+ "VISUAL": "IDENTITY"
122
+ },
123
+ "observation_delta_indices": null,
124
+ "output_features": {
125
+ "action": {
126
+ "shape": [
127
+ 7
128
+ ],
129
+ "type": "ACTION"
130
+ }
131
+ },
132
+ "pretrained_path": "physical-intelligence/pi0_base",
133
+ "resize_imgs_with_padding": [
134
+ 224,
135
+ 224
136
+ ],
137
+ "reward_delta_indices": null,
138
+ "robot_state_feature": {
139
+ "shape": [
140
+ 8
141
+ ],
142
+ "type": "STATE"
143
+ },
144
+ "tokenizer_max_length": 512,
145
+ "torch_dtype": "float32",
146
+ "train_expert_only": false,
147
+ "train_state_proj": true,
148
+ "transformers_version": "4.48.1",
149
+ "use_cache": true
150
+ }
model-00001-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca98e81632f3dd64df720bfb2bcdffa51054e87ab20fa25d4700302285a3fe7b
3
+ size 4952905824
model-00002-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f14851424d3ee71746df8c453619581c16a18e0bb3b2430d0d41c2ae574235c
3
+ size 4999824600
model-00003-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b24fc248a519818a578a101ecd0a9ab3d2a0c59783d8fda96821a4372b83684
3
+ size 4052889144
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff