Upload folder using huggingface_hub

config.json

@@ -21,7 +21,7 @@
       640
     ],
     "observation.state": [
-      6
+      7
     ]
   },
   "kl_weight": 10.0,
@@ -37,7 +37,7 @@
   },
   "output_shapes": {
     "action": [
-      6
+      7
     ]
   },
   "pre_norm": false,

config.yaml

@@ -2,7 +2,7 @@ resume: false
 device: cuda
 use_amp: false
 seed: 1000
-dataset_repo_id: satvikahuja/so100_test2
+dataset_repo_id: satvikahuja/orange_mixer_1
 video_backend: pyav
 training:
   offline_steps: 80000
@@ -169,8 +169,8 @@ fps: 30
 env:
   name: real_world
   task: null
-  state_dim: 6
-  action_dim: 6
+  state_dim: 7
+  action_dim: 7
   fps: ${fps}
 override_dataset_stats:
   observation.images.laptop:

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bba27fe8efc3543d3c1dc392e2f67f193574ce2a501d240d01891e2c5d88ef04
-size 206701072
+oid sha256:2ec4ad59e9c5d37ddbfd23627b5f82f0622517e7f84f19bd6e7a1f3067ffee44
+size 206709292