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

Signed-off-by: yuwenzho <yuwen.zhou@intel.com>
intel · Mar 10, 2024 · b0dc0de · b0dc0de
1 parent 3a254e9
commit b0dc0de
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/...nnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/prepare_model.py b/...nnxrt/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/...nnxrt/nlp/huggingface_model/text_classification/quantization/ptq_dynamic/requirements.txt b/...nnxrt/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/...onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_static/prepare_model.py b/...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/...onnxrt/nlp/huggingface_model/text_classification/quantization/ptq_static/requirements.txt b/...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