Upload modeling_prismatic.py
Browse files- modeling_prismatic.py +1 -0
modeling_prismatic.py
CHANGED
|
@@ -508,6 +508,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
|
|
| 508 |
) -> np.ndarray:
|
| 509 |
|
| 510 |
print("YESSSSSSSSS")
|
|
|
|
| 511 |
exit(1)
|
| 512 |
"""Thin wrapper around .generate() that decodes predicted actions and unnormalizes them."""
|
| 513 |
# We need to add this special empty token ('') after the colon (':') token in "ASSISTANT:"
|
|
|
|
| 508 |
) -> np.ndarray:
|
| 509 |
|
| 510 |
print("YESSSSSSSSS")
|
| 511 |
+
aa=bb
|
| 512 |
exit(1)
|
| 513 |
"""Thin wrapper around .generate() that decodes predicted actions and unnormalizes them."""
|
| 514 |
# We need to add this special empty token ('') after the colon (':') token in "ASSISTANT:"
|