fix: change deps version to fit OpenVLA

pyproject.toml

@@ -10,9 +10,10 @@ dependencies = [
     "pillow>=12.0.0",
     "robosuite>=1.5.1",
     "robosuite-models>=1.0.0",
+    "timm>=0.9.10,<1.0.0",
     "torch>=2.9.1",
     "torchvision>=0.24.1",
-    "transformers>=4.57.3",
+    "transformers==4.40.1",
 ]
 
 [dependency-groups]

src/main.py

@@ -92,7 +92,9 @@ class OpenVLADemo:
         # Predict action using OpenVLA
         # Note: unnorm_key should match your robot setup
         # For simulation, we'll use raw normalized actions
-        action = self.model.predict_action(**inputs, do_sample=False)
+        action = self.model.predict_action(
+            **inputs, do_sample=False, unnorm_key="roboturk"
+        )
 
         # Action is numpy array of shape (7,)
         return action