Skip to content

GitLab

Commit 7046de29 authored by Lysandre's avatar Lysandre Committed by Victor SANH
Browse files

E231

parent 0d3aa3c0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines.py
View file @ 7046de29
...@@ -977,7 +977,7 @@ SUPPORTED_TASKS = { ...@@ -977,7 +977,7 @@ SUPPORTED_TASKS = {
"tf": TFAutoModelForQuestionAnswering if is_tf_available() else None, "tf": TFAutoModelForQuestionAnswering if is_tf_available() else None,
"pt": AutoModelForQuestionAnswering if is_torch_available() else None, "pt": AutoModelForQuestionAnswering if is_torch_available() else None,
"default": { "default": {
"model": {"pt": "distilbert-base-cased-distilled-squad", "tf": "distilbert-base-cased-distilled-squad",}, "model": {"pt": "distilbert-base-cased-distilled-squad", "tf": "distilbert-base-cased-distilled-squad"},
"config": None, "config": None,
"tokenizer": "distilbert-base-cased", "tokenizer": "distilbert-base-cased",
}, },
......
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