diff --git a/iopaint/installer.py b/iopaint/installer.py index cb6e6e3..a7f22af 100644 --- a/iopaint/installer.py +++ b/iopaint/installer.py @@ -7,4 +7,5 @@ def install(package): def install_plugins_package(): + install("onnxruntime<=1.19.2") install("rembg[cpu]") diff --git a/iopaint/runtime.py b/iopaint/runtime.py index f3eba43..cad07c0 100644 --- a/iopaint/runtime.py +++ b/iopaint/runtime.py @@ -30,6 +30,7 @@ _CANDIDATES = [ "accelerate", "iopaint", "rembg", + "onnxruntime", ] # Check once at runtime for name in _CANDIDATES: