Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -99,6 +99,7 @@ class GPT2Vision(PreTrainedModel):
         
     def generate(self, question, image, max_new_tokens=30, **kwargs):
         prompt = f"<image>\n\nQuestion: {question}\n\nAnswer:"
+        print("prompt",prompt)
         batch = {"image": [image], "text": prompt}
         encoded_batch = self.tokenize_encode(batch, self.device)
         inputs_embeds, attention_mask = self.preprocess_inputs(encoded_batch)