VictorSanh commited on
Commit
a8b0561
1 Parent(s): 2203bf6

Update modeling_siglip.py

Browse files
Files changed (1) hide show
  1. modeling_siglip.py +2 -0
modeling_siglip.py CHANGED
@@ -1203,6 +1203,7 @@ class SiglipVisionModel(SiglipPreTrainedModel):
1203
  def forward(
1204
  self,
1205
  pixel_values,
 
1206
  output_attentions: Optional[bool] = None,
1207
  output_hidden_states: Optional[bool] = None,
1208
  return_dict: Optional[bool] = None,
@@ -1233,6 +1234,7 @@ class SiglipVisionModel(SiglipPreTrainedModel):
1233
 
1234
  return self.vision_model(
1235
  pixel_values=pixel_values,
 
1236
  output_attentions=output_attentions,
1237
  output_hidden_states=output_hidden_states,
1238
  return_dict=return_dict,
 
1203
  def forward(
1204
  self,
1205
  pixel_values,
1206
+ patch_attention_mask: Optional[torch.BoolTensor] = None,
1207
  output_attentions: Optional[bool] = None,
1208
  output_hidden_states: Optional[bool] = None,
1209
  return_dict: Optional[bool] = None,
 
1234
 
1235
  return self.vision_model(
1236
  pixel_values=pixel_values,
1237
+ patch_attention_mask=patch_attention_mask,
1238
  output_attentions=output_attentions,
1239
  output_hidden_states=output_hidden_states,
1240
  return_dict=return_dict,