Update modeling_ovis2_5.py
Browse files- 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)
|