update the image size for inference

2026-06-12 14:36:13 +08:00
parent 87f2a2abfc
commit cbf0edfcae
1 changed files with 1 additions and 1 deletions
--- a/starvla_inference_server.py
+++ b/starvla_inference_server.py
@@ -129,7 +129,7 @@ class StarvlaInferenceServer:
    def inference(self, observation: dict) -> dict:
        
        img_head, state_vec, prompt = \
-            self.parse_observation(observation)
+            self.parse_observation(observation, target_size=(410, 224))
        vla_input = {
            # "batch_images": [[img_left, img_right, img_wrist]],
            "image": [img_head],