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