From bfc8f8c94d08155befa9da94337faeefe19bdcb4 Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Fri, 30 Jul 2021 12:17:44 -0400
Subject: [PATCH] [BERT examples] update default pruned model (#157)

* [BERT examples] update default pruned model

* update src/transformers pipeline file
---
 examples/huggingface-transformers/pipelines.py | 2 +-
 src/deepsparse/transformers/pipelines.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/huggingface-transformers/pipelines.py b/examples/huggingface-transformers/pipelines.py
index 3096258a91..467ecb0ac9 100644
--- 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
index 3096258a91..467ecb0ac9 100644
--- 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"
         ),
     )
 }