tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

5705900

·

verified ·

1 Parent(s): 38af5d4

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +1 -1

modeling_prismatic.py CHANGED Viewed

@@ -516,7 +516,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
         # Run VLA inference
         model_outputs = self.generate(input_ids, max_new_tokens=self.get_action_dim(unnorm_key), output_attentions=True, return_dict=True, **kwargs)
-        print(len(model_outputs))
         # Extract predicted action tokens and translate into (normalized) continuous actions
         predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()

         # Run VLA inference
         model_outputs = self.generate(input_ids, max_new_tokens=self.get_action_dim(unnorm_key), output_attentions=True, return_dict=True, **kwargs)
+        print(model_outputs.shape)
         # Extract predicted action tokens and translate into (normalized) continuous actions
         predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()