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