From 8d8970efdd0e21b54f1c82dec21e8a5eeba609a1 Mon Sep 17 00:00:00 2001 From: NielsRogge <48327001+NielsRogge@users.noreply.github.com> Date: Mon, 11 Dec 2023 09:17:02 +0100 Subject: [PATCH] [BEiT] Fix test (#27934) Fix test --- src/transformers/models/beit/modeling_beit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/beit/modeling_beit.py b/src/transformers/models/beit/modeling_beit.py index 949c529222115a..da4721656c0285 100755 --- a/src/transformers/models/beit/modeling_beit.py +++ b/src/transformers/models/beit/modeling_beit.py @@ -1349,8 +1349,8 @@ def get_input_embeddings(self): 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: """