[Summarization Pipeline]: Fix default tokenizer (#4506)

* Fix pipelines defaults bug * one liner * style

[Summarization Pipeline]: Fix default tokenizer (#4506)
* Fix pipelines defaults bug * one liner * style
ab44630d · Sam Shleifer · GitHub · 2c1ebb8b · ab44630d
Unverified Commit ab44630d authored May 22, 2020 by Sam Shleifer Committed by GitHub May 22, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 5 deletions

src/transformers/pipelines.py src/transformers/pipelines.py +1 -5

No files found.
--- 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,