Upload modeling_prismatic.py
Browse files- modeling_prismatic.py +1 -1
modeling_prismatic.py
CHANGED
|
@@ -533,7 +533,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
|
|
| 533 |
normalized_actions,
|
| 534 |
)
|
| 535 |
|
| 536 |
-
return actions
|
| 537 |
|
| 538 |
@staticmethod
|
| 539 |
def _check_unnorm_key(norm_stats: Dict[str, Dict[str, Any]], unnorm_key: Optional[str]) -> str:
|
|
|
|
| 533 |
normalized_actions,
|
| 534 |
)
|
| 535 |
|
| 536 |
+
return actions, generated_ids
|
| 537 |
|
| 538 |
@staticmethod
|
| 539 |
def _check_unnorm_key(norm_stats: Dict[str, Dict[str, Any]], unnorm_key: Optional[str]) -> str:
|