Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: diffusion
 pipeline_tag: robotics
 tags:
+- robotics
 - diffusion
 - lerobot
-- robotics
 ---
 
 # Model Card for diffusion

config.json

@@ -23,6 +23,12 @@
                 480,
                 640
             ]
+        },
+        "observation.environment_state": {
+            "type": "ENV",
+            "shape": [
+                6
+            ]
         }
     },
     "output_features": {

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0e4b907e71e85d2a1866379e2bfe062a1c04f95806b6952dc6eeb990abd2a07e
-size 1066518712
+oid sha256:4b03e7977bfa0bd3edb182907557edb8112ceecfa602a88b93624357831a7ea2
+size 1067894968

train_config.json

@@ -105,6 +105,12 @@
                     480,
                     640
                 ]
+            },
+            "observation.environment_state": {
+                "type": "ENV",
+                "shape": [
+                    6
+                ]
             }
         },
         "output_features": {