Update modeling_prismatic.py

modeling_prismatic.py

@@ -506,10 +506,6 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
     def predict_action(
         self, input_ids: Optional[torch.LongTensor] = None, unnorm_key: Optional[str] = None, **kwargs
     ) -> np.ndarray:
-        
-        print("YESSSSSSSSS")
-        aa=bb
-        exit(1)
         """Thin wrapper around .generate() that decodes predicted actions and unnormalizes them."""
         # We need to add this special empty token ('') after the colon (':') token in "ASSISTANT:"
         # in order for the predictions to match the training configuration and be accurate.