Skip to content

Commit eb4f8e7

Browse files
Merge pull request #1896 from mikel-brostrom/upgrade-onnx
upgrade onnx to solve security warning
2 parents 98eaa0f + 8644a9d commit eb4f8e7

File tree

3 files changed

+236
-235
lines changed

3 files changed

+236
-235
lines changed

boxmot/appearance/backends/onnx_backend.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,13 @@ def load_model(self, w):
1616
# ONNXRuntime will attempt to use the first provider, and if it fails or is not
1717
# available for some reason, it will fall back to the next provider in the list
1818
if self.device == "mps":
19-
self.checker.check_packages(("onnxruntime-silicon==1.16.3",))
19+
self.checker.check_packages(("onnxruntime-silicon==1.17.0",))
2020
providers = ["MPSExecutionProvider", "CPUExecutionProvider"]
2121
elif self.device == "cuda":
22-
self.checker.check_packages(("onnxruntime-gpu==1.16.3",))
22+
self.checker.check_packages(("onnxruntime-gpu==1.17.0",))
2323
providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
2424
else:
25-
self.checker.check_packages(("onnxruntime==1.16.3",))
25+
self.checker.check_packages(("onnxruntime==1.17.0",))
2626
providers = ["CPUExecutionProvider"]
2727

2828
# Load the ONNX model using onnxruntime

0 commit comments

Comments
 (0)