Skip to content

Commit 0a606cf

Browse files
committed
fix prepare model bug
Signed-off-by: yuwenzho <[email protected]>
1 parent 2e6e432 commit 0a606cf

File tree

1 file changed

+1
-1
lines changed
  • examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic

1 file changed

+1
-1
lines changed

examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/prepare_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,6 @@ def export_onnx_model(args, model):
8989

9090
if args.input_model == 'Intel/bart-large-mrpc':
9191
import os
92-
os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification bart-large-mrpc/')
92+
os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification --export_with_transformers bart-large-mrpc/')
9393
else:
9494
export_onnx_model(args, model)

0 commit comments

Comments
 (0)