fix: change deps version to fit OpenVLA

2025-12-20 06:27:49 +08:00 · 2025-12-08 23:12:26 +08:00
parent 48bb2d3258
commit f916ef0734
3 changed files with 892 additions and 822 deletions
--- a/src/main.py
+++ b/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