Skip to content

Commit b0dc0de

Browse files
authored
Fix onnxrt example bug: bart export failure (#1658)
Signed-off-by: yuwenzho <yuwen.zhou@intel.com>
1 parent 3a254e9 commit b0dc0de

File tree

4 files changed

+15
-8
lines changed

4 files changed

+15
-8
lines changed

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

+6-3
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,10 @@ def export_onnx_model(args, model):
8888
config=AutoConfig.from_pretrained(args.input_model))
8989

9090
if args.input_model == 'Intel/bart-large-mrpc':
91-
import os
92-
os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification --export_with_transformers bart-large-mrpc/')
91+
import shutil
92+
from optimum.exporters.onnx import main_export
93+
94+
main_export(args.input_model, output="bart-large-mrpc", task="text-classification")
95+
shutil.move("bart-large-mrpc/model.onnx", args.output_model)
9396
else:
94-
export_onnx_model(args, model)
97+
export_onnx_model(args, model)

examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/requirements.txt

+2-1
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,5 @@ sympy
88
onnxruntime-extensions; python_version < '3.11'
99
numpy==1.23.5
1010
sentencepiece
11-
protobuf<=3.20.3
11+
protobuf<=3.20.3
12+
optimum

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

+6-3
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,10 @@ def export_onnx_model(args, model):
8787
config=AutoConfig.from_pretrained(args.input_model))
8888

8989
if args.input_model == 'Intel/bart-large-mrpc':
90-
import os
91-
os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification --export_with_transformers bart-large-mrpc/')
90+
import shutil
91+
from optimum.exporters.onnx import main_export
92+
93+
main_export(args.input_model, output="bart-large-mrpc", task="text-classification")
94+
shutil.move("bart-large-mrpc/model.onnx", args.output_model)
9295
else:
93-
export_onnx_model(args, model)
96+
export_onnx_model(args, model)

examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_static/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,4 @@ onnxruntime-extensions; python_version < '3.11'
99
numpy==1.23.5
1010
sentencepiece
1111
protobuf<=3.20.3
12-
optimum[exporters]
12+
optimum

0 commit comments

Comments
 (0)