Skip to content

GitLab

Unverified Commit ab44630d authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

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

* Fix pipelines defaults bug

* one liner

* style
parent 2c1ebb8b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 5 deletions
src/transformers/pipelines.py
View file @ ab44630d
...@@ -1628,11 +1628,7 @@ SUPPORTED_TASKS = { ...@@ -1628,11 +1628,7 @@ SUPPORTED_TASKS = {
"impl": SummarizationPipeline, "impl": SummarizationPipeline,
"tf": TFAutoModelWithLMHead if is_tf_available() else None, "tf": TFAutoModelWithLMHead if is_tf_available() else None,
"pt": AutoModelWithLMHead if is_torch_available() else None, "pt": AutoModelWithLMHead if is_torch_available() else None,
"default": { "default": {"model": {"pt": "bart-large-cnn", "tf": "t5-small"}, "config": None, "tokenizer": None},
"model": {"pt": "bart-large-cnn", "tf": "t5-small"},
"config": None,
"tokenizer": {"pt": ("bart-large-cnn", {"use_fast": False}), "tf": "t5-small"},
},
}, },
"translation_en_to_fr": { "translation_en_to_fr": {
"impl": TranslationPipeline, "impl": TranslationPipeline,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment