⚠️ [VitDet] Fix test (#27832)

Address test

⚠️ [VitDet] Fix test (#27832)
Address test
28e2887a · NielsRogge · GitHub · b242d0f2 · 28e2887a
Unverified Commit 28e2887a authored Dec 05, 2023 by NielsRogge Committed by GitHub Dec 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/vitdet/modeling_vitdet.py src/transformers/models/vitdet/modeling_vitdet.py +1 -1

No files found.
--- a/src/transformers/models/vitdet/modeling_vitdet.py
+++ b/src/transformers/models/vitdet/modeling_vitdet.py
@@ -815,8 +815,8 @@ class VitDetBackbone(VitDetPreTrainedModel, BackboneMixin):
    def forward(
        self,
        pixel_values: torch.Tensor,
-        output_attentions: Optional[bool] = None,
        output_hidden_states: Optional[bool] = None,
+        output_attentions: Optional[bool] = None,
        return_dict: Optional[bool] = None,
    ) -> BackboneOutput:
        """