Hugging Face's logo

ISdept
/
transformer-diffusion

Safetensors
Model card Files
xet
 Community
ISdept commited on
Commit
36b5f31
·
verified ·
1 Parent(s): 288e3a2

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. model.safetensors +1 -1
  2. policy_postprocessor.json +32 -0
  3. policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  4. policy_preprocessor.json +83 -0
  5. policy_preprocessor_step_4_normalizer_processor.safetensors +3 -0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4803e6705e5b2444e35973a6c01cb738c8ac329b5530de1591aa8f289e266a03
3
  size 166341628
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0903ff6aa328740f6f79e9577dc94b63b73db1cf4cc0dab3812c6d75c552525
3
  size 166341628
policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 7
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "IDENTITY",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:712296ed1f09e23973ac442ca6ecd054d16d578964c98f282a87dcd93966a93f
3
+ size 4400
policy_preprocessor.json ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "grid_overlay_processor",
16
+ "config": {}
17
+ },
18
+ {
19
+ "registry_name": "device_processor",
20
+ "config": {
21
+ "device": "cuda",
22
+ "float_dtype": null
23
+ }
24
+ },
25
+ {
26
+ "registry_name": "normalizer_processor",
27
+ "config": {
28
+ "eps": 1e-08,
29
+ "features": {
30
+ "observation.state": {
31
+ "type": "STATE",
32
+ "shape": [
33
+ 7
34
+ ]
35
+ },
36
+ "observation.box": {
37
+ "type": "STATE",
38
+ "shape": [
39
+ 6,
40
+ 6
41
+ ]
42
+ },
43
+ "observation.images.gripper": {
44
+ "type": "VISUAL",
45
+ "shape": [
46
+ 3,
47
+ 400,
48
+ 640
49
+ ]
50
+ },
51
+ "observation.images.front": {
52
+ "type": "VISUAL",
53
+ "shape": [
54
+ 3,
55
+ 400,
56
+ 640
57
+ ]
58
+ },
59
+ "observation.images.right": {
60
+ "type": "VISUAL",
61
+ "shape": [
62
+ 3,
63
+ 400,
64
+ 640
65
+ ]
66
+ },
67
+ "action": {
68
+ "type": "ACTION",
69
+ "shape": [
70
+ 7
71
+ ]
72
+ }
73
+ },
74
+ "norm_map": {
75
+ "VISUAL": "IDENTITY",
76
+ "STATE": "MEAN_STD",
77
+ "ACTION": "MEAN_STD"
78
+ }
79
+ },
80
+ "state_file": "policy_preprocessor_step_4_normalizer_processor.safetensors"
81
+ }
82
+ ]
83
+ }
policy_preprocessor_step_4_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:712296ed1f09e23973ac442ca6ecd054d16d578964c98f282a87dcd93966a93f
3
+ size 4400