Update modeling_internimage.py (#1)

- Update modeling_internimage.py (83c0e4aeddfaee72559e26dd22c218f23504e602)


Co-authored-by: Parakh Agarwal <[email protected]>

modeling_internimage.py

@@ -888,8 +888,8 @@ class InternImageModelForImageClassification(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor, labels=None):
-        outputs = self.model.forward(tensor)
+    def forward(self, pixel_values, labels=None):
+        outputs = self.model.forward(pixel_values)
 
         if labels is not None:
             logits = outputs['logits']