damerajee commited on
Commit
1644036
·
verified ·
1 Parent(s): 2cc3201

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py CHANGED
@@ -98,7 +98,7 @@ class GPT2Vision(PreTrainedModel):
98
 
99
 
100
  def generate(self, question, image, max_new_tokens=30, **kwargs):
101
- prompt = f"<image>\n\nQuestion: {question}\nAnswer:"
102
  print("prompt",prompt)
103
  batch = {"image": [image], "text": prompt}
104
  encoded_batch = self.tokenize_encode(batch, self.device)
 
98
 
99
 
100
  def generate(self, question, image, max_new_tokens=30, **kwargs):
101
+ prompt = f"\n\n<image>Question: {question}\nAnswer:"
102
  print("prompt",prompt)
103
  batch = {"image": [image], "text": prompt}
104
  encoded_batch = self.tokenize_encode(batch, self.device)