[BEiT] Fix test (#27934)

Fix test

[BEiT] Fix test (#27934)
Fix test
8d8970ef · NielsRogge · GitHub · 235be085 · 8d8970ef
Unverified Commit 8d8970ef authored Dec 11, 2023 by NielsRogge Committed by GitHub Dec 11, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/beit/modeling_beit.py src/transformers/models/beit/modeling_beit.py +1 -1

No files found.
--- a/src/transformers/models/beit/modeling_beit.py
+++ b/src/transformers/models/beit/modeling_beit.py
@@ -1349,8 +1349,8 @@ class BeitBackbone(BeitPreTrainedModel, BackboneMixin):
    def forward(
        self,
        pixel_values: Tensor,
-        output_attentions: Optional[bool] = None,
        output_hidden_states: Optional[bool] = None,
+        output_attentions: Optional[bool] = None,
        return_dict: Optional[bool] = None,
    ) -> BackboneOutput:
        """