From 9b95f7d07c1bea08d3e9017c3709d4e4742ee440 Mon Sep 17 00:00:00 2001 From: Lewis Tunstall Date: Mon, 28 Mar 2022 15:40:49 +0200 Subject: [PATCH] Remove kwargs argument from IBERT MLM forward pass --- src/transformers/models/ibert/modeling_ibert.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/transformers/models/ibert/modeling_ibert.py b/src/transformers/models/ibert/modeling_ibert.py index d82aed83b622..6e8821cfafb8 100644 --- a/src/transformers/models/ibert/modeling_ibert.py +++ b/src/transformers/models/ibert/modeling_ibert.py @@ -893,7 +893,6 @@ def forward( output_attentions: Optional[bool] = None, output_hidden_states: Optional[bool] = None, return_dict: Optional[bool] = None, - **kwargs, ) -> Union[MaskedLMOutput, Tuple[torch.FloatTensor]]: r""" labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):