Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -24,8 +24,8 @@ class GPT2Vision(PreTrainedModel):
     def device(self):
         return self.text_model.device
 
-    def encode_image(self, image):
-        return self.vision_encoder(image)
+    def encode_image(self, image,device):
+        return self.vision_encoder(image,device=device)
 
     def input_embeds(self, prompt, image_embeds, tokenizer):
         def _tokenize(txt):