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

yuwenzho · web-flow · commit b0dc0de8325e · 2024-03-10T17:37:54.000+08:00
Signed-off-by: yuwenzho &lt;yuwen.zhou@intel.com&gt;
diff --git a/examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/prepare_model.py b/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)
diff --git a/examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/requirements.txt b/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
diff --git a/examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_static/prepare_model.py b/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)
diff --git a/examples/onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_static/requirements.txt b/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
