SummarizationPipeline: init required task name (#5086)

* SummarizationPipeline: init required task name

* Update src/transformers/pipelines.py

Co-authored-by: Sam Shleifer <sshleifer@gmail.com>

* Apply suggestions from code review

Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
This commit is contained in:
Julien Chaumond 2020-06-20 09:16:30 +02:00 коммит произвёл GitHub
Родитель 2fd28d4363
Коммит aa6a29bc25
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -1391,6 +1391,10 @@ class SummarizationPipeline(Pipeline):
on the associated CUDA device id.
"""
def __init__(self, **kwargs):
kwargs.update(task="summarization")
super().__init__(**kwargs)
def __call__(
self, *documents, return_tensors=False, return_text=True, clean_up_tokenization_spaces=False, **generate_kwargs
):