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" ), ) }