Convert VLA-JEPA-SimplerEnv.pt to safetensors

config.json

@@ -71,6 +71,7 @@
     "repeated_diffusion_steps": 8,
     "resize_images_to": null,
     "binarize_gripper_action": true,
+    "pre_snap_gripper_action": true,
     "clip_normalized_actions": true,
     "torch_dtype": "bfloat16",
     "optimizer_lr": 0.0001,

policy_postprocessor.json

@@ -5,6 +5,10 @@
       "registry_name": "vla_jepa_clip_actions",
       "config": {}
     },
+    {
+      "registry_name": "vla_jepa_pre_snap_gripper",
+      "config": {}
+    },
     {
       "registry_name": "unnormalizer_processor",
       "config": {
@@ -31,7 +35,7 @@
           "ACTION": "MIN_MAX"
         }
       },
-      "state_file": "policy_postprocessor_step_1_unnormalizer_processor.safetensors"
+      "state_file": "policy_postprocessor_step_2_unnormalizer_processor.safetensors"
     },
     {
       "registry_name": "vla_jepa_binarize_gripper",

policy_postprocessor_step_2_unnormalizer_processor.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00e22318f95221ba2798bdd6e0f84a1a51a02d55699a90a552fc474d7e6c6939
+size 668