From 02acff7859030229a3ce49b641d47b5b6e0bcbd5 Mon Sep 17 00:00:00 2001 From: Eric Mintun Date: Sat, 15 Apr 2023 10:34:28 -0700 Subject: [PATCH] Fix lint. --- scripts/export_onnx_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/export_onnx_model.py b/scripts/export_onnx_model.py index 8441258..5c6f838 100644 --- a/scripts/export_onnx_model.py +++ b/scripts/export_onnx_model.py @@ -161,7 +161,7 @@ def run_export( if onnxruntime_exists: ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()} # set cpu provider default - providers = ['CPUExecutionProvider'] + providers = ["CPUExecutionProvider"] ort_session = onnxruntime.InferenceSession(output, providers=providers) _ = ort_session.run(None, ort_inputs) print("Model has successfully been run with ONNXRuntime.")