Upload folder using huggingface_hub

README.md

@@ -1,14 +1,14 @@
 ---
 base_model: lerobot/smolvla_base
-datasets: thewisp/move-cube-v2
+datasets: unknown
 library_name: lerobot
 license: apache-2.0
 model_name: smolvla
 pipeline_tag: robotics
 tags:
 - robotics
-- smolvla
 - lerobot
+- smolvla
 ---
 
 # Model Card for smolvla

config.json

@@ -1,11 +1,6 @@
 {
     "type": "smolvla",
     "n_obs_steps": 5,
-    "normalization_mapping": {
-        "VISUAL": "IDENTITY",
-        "STATE": "MEAN_STD",
-        "ACTION": "MEAN_STD"
-    },
     "input_features": {
         "observation.state": {
             "type": "STATE",
@@ -45,8 +40,14 @@
     "private": null,
     "tags": null,
     "license": null,
+    "pretrained_path": null,
     "chunk_size": 50,
     "n_action_steps": 50,
+    "normalization_mapping": {
+        "VISUAL": "IDENTITY",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
     "max_state_dim": 32,
     "max_action_dim": 32,
     "resize_imgs_with_padding": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d27d721093379a0c18c42d5da917a837718459d27a32b0a362ad59343888c515
-size 906713296
+oid sha256:568a18829ced3c7aea4050fa1ca4f9d70f9f22635c52a161ae4dc5e01137165c
+size 1800257992

policy_postprocessor.json

@@ -0,0 +1,32 @@
+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "IDENTITY",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a7c4bffcb2bc1085bc5c891bfc5b881ed1c08fa40466bb80111152d078353db
+size 384

policy_preprocessor.json

@@ -0,0 +1,79 @@
+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "smolvla_new_line_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "tokenizer_processor",
+      "config": {
+        "max_length": 48,
+        "task_key": "task",
+        "padding_side": "right",
+        "padding": "max_length",
+        "truncation": true,
+        "tokenizer_name": "HuggingFaceTB/SmolVLM2-500M-Video-Instruct"
+      }
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.top": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.wrist": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "IDENTITY",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
+    }
+  ]
+}

policy_preprocessor_step_5_normalizer_processor.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a7c4bffcb2bc1085bc5c891bfc5b881ed1c08fa40466bb80111152d078353db
+size 384