[BERT examples] update default pruned model (#157)

* [BERT examples] update default pruned model * update src/transformers pipeline file
neuralmagic · Jul 30, 2021 · bfc8f8c · bfc8f8c
1 parent de70f82
commit bfc8f8c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/huggingface-transformers/pipelines.py b/examples/huggingface-transformers/pipelines.py
@@ -814,7 +814,7 @@ class TaskInfo:
         ),
         default_pruned_stub=(
             "zoo:nlp/question_answering/bert-base/pytorch/huggingface/squad/"
-            "pruned-moderate"
+            "pruned-aggressive_98"
         ),
     )
 }

diff --git a/src/deepsparse/transformers/pipelines.py b/src/deepsparse/transformers/pipelines.py
@@ -814,7 +814,7 @@ class TaskInfo:
         ),
         default_pruned_stub=(
             "zoo:nlp/question_answering/bert-base/pytorch/huggingface/squad/"
-            "pruned-moderate"
+            "pruned-aggressive_98"
         ),
     )
 }