tt1225 commited on
Commit
6fca09e
·
verified ·
1 Parent(s): 268e7ba

Update modeling_prismatic.py

Browse files
Files changed (1) hide show
  1. modeling_prismatic.py +1 -1
modeling_prismatic.py CHANGED
@@ -522,7 +522,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
522
 
523
  b = model_outputs["past_key_values"]
524
  for i in b:
525
- print(i.keys())
526
  # print(len(b))
527
 
528
  # Extract predicted action tokens and translate into (normalized) continuous actions
 
522
 
523
  b = model_outputs["past_key_values"]
524
  for i in b:
525
+ print(len(i))
526
  # print(len(b))
527
 
528
  # Extract predicted action tokens and translate into (normalized) continuous actions