From 59b9351b785b231814c462b62da8ce1dd9ed928b Mon Sep 17 00:00:00 2001 From: Samuel Larkin Date: Thu, 23 Mar 2023 11:14:11 -0400 Subject: [PATCH] Minor typo in pipeline FillMaskPipeline's documentation. (#22339) --- src/transformers/pipelines/fill_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/fill_mask.py b/src/transformers/pipelines/fill_mask.py index fd977a72f8..74992485d3 100644 --- a/src/transformers/pipelines/fill_mask.py +++ b/src/transformers/pipelines/fill_mask.py @@ -234,7 +234,7 @@ class FillMaskPipeline(Pipeline): - **sequence** (`str`) -- The corresponding input with the mask token prediction. - **score** (`float`) -- The corresponding probability. - **token** (`int`) -- The predicted token id (to replace the masked one). - - **token** (`str`) -- The predicted token (to replace the masked one). + - **token_str** (`str`) -- The predicted token (to replace the masked one). """ outputs = super().__call__(inputs, **kwargs) if isinstance(inputs, list) and len(inputs) == 1: