Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -74,9 +74,9 @@ class GPT2Vision(PreTrainedModel):
         }
 
     def preprocess_inputs(self, batch):
-        pixel_values = batch['pixel_values'].squeeze(1)
-        input_ids = batch['input_ids'].squeeze(1)
-        attention_mask = batch['attention_mask'].squeeze(1)
+        pixel_values = batch['pixel_values']
+        input_ids = batch['input_ids']
+        attention_mask = batch['attention_mask']
         input_ids = input_ids.to(self.device)
         attention_mask = attention_mask.to(self.device)
         pixel_values = pixel_values.to(self.device)