runninglsy commited on
Commit
b44426e
·
verified ·
1 Parent(s): 3f51754

Update modeling_ovis2_5.py

Browse files
Files changed (1) hide show
  1. modeling_ovis2_5.py +0 -1
modeling_ovis2_5.py CHANGED
@@ -910,7 +910,6 @@ class Ovis2_5(OvisPreTrainedModel):
910
  # check if the thinking process has finished (151668 is </think>)
911
  # and prepare the second model input
912
  if 151668 not in output_ids_list:
913
- print("thinking budget is reached")
914
  early_stopping_text = "\n\nConsidering the limited time by the user, I have to give the solution based on the thinking directly now.\n</think>\n\n"
915
  early_stopping_ids = self.text_tokenizer(early_stopping_text, return_tensors="pt", return_attention_mask=False).input_ids.to(inputs.device)
916
  input_ids_appendent = torch.cat([output_ids, early_stopping_ids], dim=-1)
 
910
  # check if the thinking process has finished (151668 is </think>)
911
  # and prepare the second model input
912
  if 151668 not in output_ids_list:
 
913
  early_stopping_text = "\n\nConsidering the limited time by the user, I have to give the solution based on the thinking directly now.\n</think>\n\n"
914
  early_stopping_ids = self.text_tokenizer(early_stopping_text, return_tensors="pt", return_attention_mask=False).input_ids.to(inputs.device)
915
  input_ids_appendent = torch.cat([output_ids, early_stopping_ids], dim=-1)