Update modeling_prismatic.py

modeling_prismatic.py

@@ -520,11 +520,14 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
         print(model_outputs.keys())
 
         a = model_outputs["sequences"]
-        # print(a.key
+        print(a.shape)
 
         b = model_outputs["past_key_values"]
         for i in b:
             print(len(i))
+            j, k = i
+            print("j:", j.shape)
+            print("k:", k.shape)
         # print(len(b))
 
         # Extract predicted action tokens and translate into (normalized) continuous actions