czczup parakh01 commited on
Commit
7b940a2
·
verified ·
1 Parent(s): db73967

Update modeling_internimage.py (#1)

Browse files

- Update modeling_internimage.py (83c0e4aeddfaee72559e26dd22c218f23504e602)


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

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