Update modeling_prismatic.py
Browse files- modeling_prismatic.py +3 -1
modeling_prismatic.py
CHANGED
|
@@ -521,7 +521,9 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
|
|
| 521 |
# print(a.key
|
| 522 |
|
| 523 |
b = model_outputs["past_key_values"]
|
| 524 |
-
|
|
|
|
|
|
|
| 525 |
|
| 526 |
# Extract predicted action tokens and translate into (normalized) continuous actions
|
| 527 |
predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()
|
|
|
|
| 521 |
# print(a.key
|
| 522 |
|
| 523 |
b = model_outputs["past_key_values"]
|
| 524 |
+
for i in b:
|
| 525 |
+
print(b.keys())
|
| 526 |
+
# print(len(b))
|
| 527 |
|
| 528 |
# Extract predicted action tokens and translate into (normalized) continuous actions
|
| 529 |
predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()
|