Skip to content

Commit bfc8f8c

Browse files
authored
[BERT examples] update default pruned model (#157)
* [BERT examples] update default pruned model * update src/transformers pipeline file
1 parent de70f82 commit bfc8f8c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/huggingface-transformers/pipelines.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -814,7 +814,7 @@ class TaskInfo:
814814
),
815815
default_pruned_stub=(
816816
"zoo:nlp/question_answering/bert-base/pytorch/huggingface/squad/"
817-
"pruned-moderate"
817+
"pruned-aggressive_98"
818818
),
819819
)
820820
}

src/deepsparse/transformers/pipelines.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -814,7 +814,7 @@ class TaskInfo:
814814
),
815815
default_pruned_stub=(
816816
"zoo:nlp/question_answering/bert-base/pytorch/huggingface/squad/"
817-
"pruned-moderate"
817+
"pruned-aggressive_98"
818818
),
819819
)
820820
}

0 commit comments

Comments
 (0)