Fix onnxruntime load error for newest onnxruntime

6abd0325 · triple-Mu · ericmintun · dec4c129 · 6abd0325
Commit 6abd0325 authored Apr 06, 2023 by triple-Mu Committed by ericmintun Apr 12, 2023
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

scripts/export_onnx_model.py scripts/export_onnx_model.py +3 -1

No files found.
--- a/scripts/export_onnx_model.py
+++ b/scripts/export_onnx_model.py
@@ -160,7 +160,9 @@ def run_export(

    if onnxruntime_exists:
        ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()}
-        ort_session = onnxruntime.InferenceSession(output)
+        # set cpu provider default
+        providers = ['CPUExecutionProvider']
+        ort_session = onnxruntime.InferenceSession(output, providers=providers)
        _ = ort_session.run(None, ort_inputs)
        print("Model has successfully been run with ONNXRuntime.")