tt1225
/

openvla-7b-devel

Image-Text-to-Text

Model card Files Files and versions

tt1225 commited on Apr 29

Commit

452b54c

·

verified ·

1 Parent(s): 38474ea

Update modeling_prismatic.py

Files changed (1) hide show

modeling_prismatic.py +1 -1

modeling_prismatic.py CHANGED Viewed

@@ -521,7 +521,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
         # print(a.key
         b = model_outputs["past_key_values"]
-        print(b.keys())
         # 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()

         # print(a.key
         b = model_outputs["past_key_values"]
+        print(len(b))
         # 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()