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:
|