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

