Weiyun1025 commited on
Commit
9b9d077
·
verified ·
1 Parent(s): 2b75d5f

Upload modeling_internvl_chat.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_internvl_chat.py +1 -1
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,