pepijn223 HF Staff commited on
Commit
2e9eeaa
·
verified ·
1 Parent(s): 2687b14

add processors

Browse files
policy_postprocessor.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {},
9
+ "norm_map": {
10
+ "VISUAL": "IDENTITY",
11
+ "STATE": "QUANTILES",
12
+ "ACTION": "MEAN_STD"
13
+ }
14
+ },
15
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
16
+ },
17
+ {
18
+ "registry_name": "device_processor",
19
+ "config": {
20
+ "device": "cpu",
21
+ "float_dtype": null
22
+ }
23
+ }
24
+ ]
25
+ }
policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4437ecae9394b31283af54f2a72ab4188e2bf4818dc6169e71c51740d3aba8f
3
+ size 816
policy_preprocessor.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "pi05_prepare_state_tokenizer_processor_step",
16
+ "config": {}
17
+ },
18
+ {
19
+ "registry_name": "tokenizer_processor",
20
+ "config": {
21
+ "max_length": 200,
22
+ "task_key": "task",
23
+ "padding_side": "right",
24
+ "padding": "max_length",
25
+ "truncation": true,
26
+ "tokenizer_name": "google/paligemma-3b-pt-224"
27
+ }
28
+ },
29
+ {
30
+ "registry_name": "device_processor",
31
+ "config": {
32
+ "device": "cuda",
33
+ "float_dtype": null
34
+ }
35
+ },
36
+ {
37
+ "registry_name": "normalizer_processor",
38
+ "config": {
39
+ "eps": 1e-08,
40
+ "features": {},
41
+ "norm_map": {
42
+ "VISUAL": "IDENTITY",
43
+ "STATE": "QUANTILES",
44
+ "ACTION": "MEAN_STD"
45
+ }
46
+ },
47
+ "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
48
+ }
49
+ ]
50
+ }
policy_preprocessor_step_5_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4437ecae9394b31283af54f2a72ab4188e2bf4818dc6169e71c51740d3aba8f
3
+ size 816