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