From 11f7d3cb79d13542f81e8e7653b3ee0b8eccceb2 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Tue, 18 Mar 2025 09:49:44 +0800
Subject: [PATCH] fix onnxruntime version

fix onnxruntime bug on windows: DLL load failed while importing onnxruntime_pybind11_state
---
 iopaint/installer.py | 1 +
 iopaint/runtime.py   | 1 +
 2 files changed, 2 insertions(+)

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: