Update modeling_internimage.py

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']