styling

0d3aa3c0 · VictorSanh · Victor SANH · d8b43600 · 0d3aa3c0
Commit 0d3aa3c0 authored Feb 07, 2020 by VictorSanh Committed by Victor SANH Feb 07, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

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

No files found.
--- a/src/transformers/pipelines.py
+++ b/src/transformers/pipelines.py
@@ -977,10 +977,7 @@ SUPPORTED_TASKS = {
        "tf": TFAutoModelForQuestionAnswering if is_tf_available() else None,
        "pt": AutoModelForQuestionAnswering if is_torch_available() else None,
        "default": {
-            "model": {
+            "model": {"pt": "distilbert-base-cased-distilled-squad", "tf": "distilbert-base-cased-distilled-squad",},
-                "pt": "distilbert-base-cased-distilled-squad",
-                "tf": "distilbert-base-cased-distilled-squad",
-            },
            "config": None,
            "tokenizer": "distilbert-base-cased",
        },