parakh01 commited on
Commit
83c0e4a
·
verified ·
1 Parent(s): db73967

Update modeling_internimage.py

Browse files

Needed for the forward pass in run_image_classification.

Files changed (1) hide show
  1. modeling_internimage.py +2 -2
modeling_internimage.py CHANGED
@@ -888,8 +888,8 @@ class InternImageModelForImageClassification(PreTrainedModel):
888
  remove_center=config.remove_center, # for InternImage-H/G
889
  )
890
 
891
- def forward(self, tensor, labels=None):
892
- outputs = self.model.forward(tensor)
893
 
894
  if labels is not None:
895
  logits = outputs['logits']
 
888
  remove_center=config.remove_center, # for InternImage-H/G
889
  )
890
 
891
+ def forward(self, pixel_values, labels=None):
892
+ outputs = self.model.forward(pixel_values)
893
 
894
  if labels is not None:
895
  logits = outputs['logits']