From 3b583d02d6d63295e1756116cd611d651c7ffe96 Mon Sep 17 00:00:00 2001 From: Eunhyuk Shin Date: Mon, 8 Mar 2021 19:10:03 +0900 Subject: [PATCH] Fix typo in docstring for pipeline (#10591) --- src/transformers/pipelines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index 33c6a42947..05e28b4d5c 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -271,7 +271,7 @@ def pipeline( - :obj:`"text2text-generation"`: will return a :class:`~transformers.Text2TextGenerationPipeline`. - :obj:`"text-generation"`: will return a :class:`~transformers.TextGenerationPipeline`. - :obj:`"zero-shot-classification:`: will return a :class:`~transformers.ZeroShotClassificationPipeline`. - - :obj:`"conversation"`: will return a :class:`~transformers.ConversationalPipeline`. + - :obj:`"conversational"`: will return a :class:`~transformers.ConversationalPipeline`. model (:obj:`str` or :obj:`~transformers.PreTrainedModel` or :obj:`~transformers.TFPreTrainedModel`, `optional`): The model that will be used by the pipeline to make predictions. This can be a model identifier or an actual instance of a pretrained model inheriting from :class:`~transformers.PreTrainedModel` (for PyTorch)