Update modeling_prismatic.py

modeling_prismatic.py

@@ -515,6 +515,12 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
         model_outputs = self.generate(input_ids, max_new_tokens=self.get_action_dim(unnorm_key), **kwargs)
         print(model_outputs.keys())
 
+        a = model_outputs["sequences"]
+        print(a.keys())
+
+        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()
         discretized_actions = self.vocab_size - predicted_action_token_ids