Skip to content

GitLab

Unverified Commit 8d8970ef authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

[BEiT] Fix test (#27934) 

Fix test
parent 235be085
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/beit/modeling_beit.py
View file @ 8d8970ef
...@@ -1349,8 +1349,8 @@ class BeitBackbone(BeitPreTrainedModel, BackboneMixin): ...@@ -1349,8 +1349,8 @@ class BeitBackbone(BeitPreTrainedModel, BackboneMixin):
def forward( def forward(
self, self,
pixel_values: Tensor, pixel_values: Tensor,
output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
output_attentions: Optional[bool] = None,
return_dict: Optional[bool] = None, return_dict: Optional[bool] = None,
) -> BackboneOutput: ) -> BackboneOutput:
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment