⚠️ [VitDet] Fix test (#27832)

Address test
huggingface · Dec 5, 2023 · 28e2887 · 28e2887
1 parent b242d0f
commit 28e2887
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/vitdet/modeling_vitdet.py b/src/transformers/models/vitdet/modeling_vitdet.py
@@ -815,8 +815,8 @@ def get_input_embeddings(self) -> VitDetEmbeddings:
     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:
         """