Merge pull request #55 from vespa-engine/lesters/specify-onnx-executi…

…on-provider
vespa-engine · Sep 21, 2023 · 6d879ea · 6d879ea
2 parents 162e140 + 499ab58
commit 6d879ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/learntorank/test_ml.py b/learntorank/test_ml.py
@@ -111,7 +111,7 @@ def _predict_with_onnx(onnx_file_path, model_inputs):
         os.environ[
             "KMP_DUPLICATE_LIB_OK"
         ] = "True"  # required to run on mac https://stackoverflow.com/a/53014308
-        m = InferenceSession(onnx_file_path)
+        m = InferenceSession(onnx_file_path, providers=['CPUExecutionProvider'])
         (out,) = m.run(input_feed=model_inputs, output_names=["output_0"])
         return out