Update modeling_prismatic.py
Browse files- 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
|
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
|