tt1225 commited on
Commit
b492183
·
verified ·
1 Parent(s): a4e0794

Upload modeling_prismatic.py

Browse files
Files changed (1) hide show
  1. 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: