Skip to content

GitLab

Unverified Commit c85547af authored by lewtun's avatar lewtun Committed by GitHub
Browse files

Remove kwargs argument from IBERT MLM forward pass (#16449)

parent da936942
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/models/ibert/modeling_ibert.py
View file @ c85547af
...@@ -893,7 +893,6 @@ class IBertForMaskedLM(IBertPreTrainedModel): ...@@ -893,7 +893,6 @@ class IBertForMaskedLM(IBertPreTrainedModel):
output_attentions: Optional[bool] = None, output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
return_dict: Optional[bool] = None, return_dict: Optional[bool] = None,
**kwargs,
) -> Union[MaskedLMOutput, Tuple[torch.FloatTensor]]: ) -> Union[MaskedLMOutput, Tuple[torch.FloatTensor]]:
r""" r"""
labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*): labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment