Update huggingface model code

classification/huggingface/22k_model/internimage_g_jointto22k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/22k_model/internimage_h_jointto22k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/22k_model/internimage_l_22k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/22k_model/internimage_xl_22k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_b_1k_224/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_g_22kto1k_512/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_h_22kto1k_640/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_l_22kto1k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_s_1k_224/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_t_1k_224/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']

classification/huggingface/in1k_model/internimage_xl_22kto1k_384/modeling_internimage.py

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
 
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 
 
 class InternImageModelForImageClassification(PreTrainedModel):
@@ -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']