Upload modeling_internvl_chat.py with huggingface_hub
Browse files
    	
        modeling_internvl_chat.py
    CHANGED
    
    | 
         @@ -240,7 +240,7 @@ class InternVLChatModel(PreTrainedModel): 
     | 
|
| 240 | 
         
             
                    model_inputs = tokenizer(queries, return_tensors='pt', padding=True)
         
     | 
| 241 | 
         
             
                    input_ids = model_inputs['input_ids'].cuda()
         
     | 
| 242 | 
         
             
                    attention_mask = model_inputs['attention_mask'].cuda()
         
     | 
| 243 | 
         
            -
                    eos_token_id = tokenizer.convert_tokens_to_ids(template.sep)
         
     | 
| 244 | 
         
             
                    generation_config['eos_token_id'] = eos_token_id
         
     | 
| 245 | 
         
             
                    generation_output = self.generate(
         
     | 
| 246 | 
         
             
                        pixel_values=pixel_values,
         
     | 
| 
         | 
|
| 240 | 
         
             
                    model_inputs = tokenizer(queries, return_tensors='pt', padding=True)
         
     | 
| 241 | 
         
             
                    input_ids = model_inputs['input_ids'].cuda()
         
     | 
| 242 | 
         
             
                    attention_mask = model_inputs['attention_mask'].cuda()
         
     | 
| 243 | 
         
            +
                    eos_token_id = tokenizer.convert_tokens_to_ids(template.sep.strip())
         
     | 
| 244 | 
         
             
                    generation_config['eos_token_id'] = eos_token_id
         
     | 
| 245 | 
         
             
                    generation_output = self.generate(
         
     | 
| 246 | 
         
             
                        pixel_values=pixel_values,
         
     |