Fix onnxrt example bug: bart export failure (#1658)

Signed-off-by: yuwenzho <[email protected]>

Author: yuwenzho

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

@@ -88,7 +88,10 @@ def export_onnx_model(args, model):
         config=AutoConfig.from_pretrained(args.input_model))
 
     if args.input_model == 'Intel/bart-large-mrpc':
-        import os
-        os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification --export_with_transformers bart-large-mrpc/')
+        import shutil
+        from optimum.exporters.onnx import main_export
+
+        main_export(args.input_model, output="bart-large-mrpc", task="text-classification")
+        shutil.move("bart-large-mrpc/model.onnx", args.output_model)
     else:
-        export_onnx_model(args, model)
+        export_onnx_model(args, model)

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

@@ -8,4 +8,5 @@ sympy
 onnxruntime-extensions; python_version < '3.11'
 numpy==1.23.5
 sentencepiece
-protobuf<=3.20.3
+protobuf<=3.20.3
+optimum

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

@@ -87,7 +87,10 @@ def export_onnx_model(args, model):
         config=AutoConfig.from_pretrained(args.input_model))
 
     if args.input_model == 'Intel/bart-large-mrpc':
-        import os
-        os.system('python -m transformers.onnx --model=Intel/bart-large-mrpc --feature=sequence-classification --export_with_transformers bart-large-mrpc/')
+        import shutil
+        from optimum.exporters.onnx import main_export
+
+        main_export(args.input_model, output="bart-large-mrpc", task="text-classification")
+        shutil.move("bart-large-mrpc/model.onnx", args.output_model)
     else:
-        export_onnx_model(args, model)
+        export_onnx_model(args, model)

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

@@ -9,4 +9,4 @@ onnxruntime-extensions; python_version < '3.11'
 numpy==1.23.5
 sentencepiece
 protobuf<=3.20.3
-optimum[exporters]
+optimum