VictorSanh
commited on
Commit
•
a8b0561
1
Parent(s):
2203bf6
Update modeling_siglip.py
Browse files- 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,
|