diff --git a/src/transformers/pipelines/fill_mask.py b/src/transformers/pipelines/fill_mask.py index a572a1642ab617..c14f54118486b9 100644 --- a/src/transformers/pipelines/fill_mask.py +++ b/src/transformers/pipelines/fill_mask.py @@ -245,12 +245,12 @@ def _sanitize_parameters(self, top_k=None, targets=None, tokenizer_kwargs=None): ) return preprocess_params, {}, postprocess_params - def __call__(self, inputs, *args, **kwargs): + def __call__(self, inputs, **kwargs): """ Fill the masked token in the text(s) given as inputs. Args: - args (`str` or `List[str]`): + inputs (`str` or `List[str]`): One or several texts (or one list of prompts) with masked tokens. targets (`str` or `List[str]`, *optional*): When passed, the model will limit the scores to the passed targets instead of looking up in the whole