huang11 commited on
Commit
7d0aba3
·
verified ·
1 Parent(s): 9f67b9f

fix check_model_inputs

Browse files
Files changed (1) hide show
  1. modeling_interns1_pro.py +3 -3
modeling_interns1_pro.py CHANGED
@@ -986,7 +986,7 @@ class InternS1ProTextModel(InternS1ProPreTrainedModel):
986
  # Initialize weights and apply final processing
987
  self.post_init()
988
 
989
- @check_model_inputs()
990
  @auto_docstring
991
  def forward(
992
  self,
@@ -1212,7 +1212,7 @@ class InternS1ProModel(InternS1ProPreTrainedModel):
1212
  return special_image_mask, special_video_mask
1213
 
1214
  @auto_docstring
1215
- @check_model_inputs()
1216
  def forward(
1217
  self,
1218
  input_ids: torch.LongTensor = None,
@@ -1405,7 +1405,7 @@ class InternS1ProForConditionalGeneration(InternS1ProPreTrainedModel, Generation
1405
  def visual(self):
1406
  return self.model.visual
1407
 
1408
- @check_model_inputs()
1409
  def forward(
1410
  self,
1411
  input_ids: torch.LongTensor = None,
 
986
  # Initialize weights and apply final processing
987
  self.post_init()
988
 
989
+ @check_model_inputs
990
  @auto_docstring
991
  def forward(
992
  self,
 
1212
  return special_image_mask, special_video_mask
1213
 
1214
  @auto_docstring
1215
+ @check_model_inputs
1216
  def forward(
1217
  self,
1218
  input_ids: torch.LongTensor = None,
 
1405
  def visual(self):
1406
  return self.model.visual
1407
 
1408
+ @check_model_inputs
1409
  def forward(
1410
  self,
1411
  input_ids: torch.LongTensor = None,