damerajee commited on
Commit
c62ed8f
·
verified ·
1 Parent(s): a94f8e2

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. 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
- from .vision_encoder import VisionEncoder
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.eos_token_id,
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
  )