Update modeling_gpt2vision.py
Browse files- modeling_gpt2vision.py +2 -2
modeling_gpt2vision.py
CHANGED
@@ -2,7 +2,7 @@ import torch
|
|
2 |
import torch.nn as nn
|
3 |
from transformers import PreTrainedModel, AutoTokenizer
|
4 |
from .configuration_gpt2vision import GPT2VisionConfig ,GPT2Config
|
5 |
-
|
6 |
|
7 |
IMAGE_TOKEN = "<image>"
|
8 |
ANSWER_EOS = "<|endoftext|>"
|
@@ -161,7 +161,7 @@ class GPT2Vision(PreTrainedModel):
|
|
161 |
inputs_embeds=inputs_embeds,
|
162 |
attention_mask=attention_mask,
|
163 |
max_new_tokens=max_new_tokens,
|
164 |
-
pad_token_id=self.tokenizer.
|
165 |
eos_token_id=self.tokenizer.eos_token_id,
|
166 |
**kwargs
|
167 |
)
|
|
|
2 |
import torch.nn as nn
|
3 |
from transformers import PreTrainedModel, AutoTokenizer
|
4 |
from .configuration_gpt2vision import GPT2VisionConfig ,GPT2Config
|
5 |
+
|
6 |
|
7 |
IMAGE_TOKEN = "<image>"
|
8 |
ANSWER_EOS = "<|endoftext|>"
|
|
|
161 |
inputs_embeds=inputs_embeds,
|
162 |
attention_mask=attention_mask,
|
163 |
max_new_tokens=max_new_tokens,
|
164 |
+
pad_token_id=self.tokenizer.pad_token_id,
|
165 |
eos_token_id=self.tokenizer.eos_token_id,
|
166 |
**kwargs
|
167 |
)
|