Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add onnx to installer

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-26 17:05:11 -04:00
parent 1227e76128
commit 738f115d06
3 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+2
View File
@@ -390,6 +390,8 @@ def check_python():
# check onnx version
def check_onnx():
if not installed('onnx', quiet=True):
install('onnx', 'onnx', ignore=True)
if not installed('onnxruntime', quiet=True) and not installed('onnxruntime-gpu', quiet=True): # allow either
install('onnxruntime', 'onnxruntime', ignore=True)