Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -82,15 +82,11 @@ class GPT2Vision(PreTrainedModel):
             "input_ids": encoded_input.input_ids.to(self.device),
             "attention_mask": encoded_input.attention_mask.to(self.device)
         }
-        print("batch",batch)
-        print("pixel_values",batch['pixel_values'].size())
-        print("input_ids",batch['input_ids'].size())
-        print("attention_mask",batch['attention_mask'].size())
+
         
         inputs_embeds, attention_mask, input_ids = self.preprocess_inputs(batch)
 
-        print("inputs_embeds",inputs_embeds.size())
-        print("attention_mask",attention_mask.size())
+
         
         output_sequences = self.language_model.generate(
             inputs_embeds=inputs_embeds,