From ab44630db2b45577d9c7a3478e47ad1f2e2b1e4f Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Fri, 22 May 2020 17:49:45 -0400 Subject: [PATCH] [Summarization Pipeline]: Fix default tokenizer (#4506) * Fix pipelines defaults bug * one liner * style --- src/transformers/pipelines.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py index c193145715..6b496e74c6 100755 --- a/src/transformers/pipelines.py +++ b/src/transformers/pipelines.py @@ -1628,11 +1628,7 @@ SUPPORTED_TASKS = { "impl": SummarizationPipeline, "tf": TFAutoModelWithLMHead if is_tf_available() else None, "pt": AutoModelWithLMHead if is_torch_available() else None, - "default": { - "model": {"pt": "bart-large-cnn", "tf": "t5-small"}, - "config": None, - "tokenizer": {"pt": ("bart-large-cnn", {"use_fast": False}), "tf": "t5-small"}, - }, + "default": {"model": {"pt": "bart-large-cnn", "tf": "t5-small"}, "config": None, "tokenizer": None}, }, "translation_en_to_fr": { "impl": TranslationPipeline,