dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

styling

This commit is contained in:
VictorSanh 2020-02-07 20:06:51 +00:00 committed by Victor SANH
parent d8b43600fd
commit 0d3aa3c04c
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/pipelines.py
View File

@ -977,10 +977,7 @@ SUPPORTED_TASKS = {
"tf": TFAutoModelForQuestionAnswering if is_tf_available() else None,
"pt": AutoModelForQuestionAnswering if is_torch_available() else None,
"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,
"tokenizer": "distilbert-base-cased",
},