Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -68,7 +68,7 @@ class GPT2Vision(PreTrainedModel):
 
     def generate(self, question, image, max_new_tokens=30, **kwargs):
         # Process the image
-        img_embs = self.vision_encoder(image.unsqueeze(0), device=self.device)
+        img_embs = self.vision_encoder(image, device=self.device)
         img_embs = self.mlp(img_embs)
         
         # Tokenize the question