fix rembg dependencies

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-07-21 08:20:55 +02:00
parent 1281cf8132
commit 315719f9e5
4 changed files with 18 additions and 9 deletions
+7 -3
View File
@@ -18,6 +18,12 @@ models = [
]
def dependencies():
from installer import install
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg", "numba"]:
install(pkg, no_deps=True, ignore=False)
class ScriptPostprocessingRembg(scripts_postprocessing.ScriptPostprocessing):
name = "Remove background"
order = 20000
@@ -81,9 +87,7 @@ class ScriptPostprocessingRembg(scripts_postprocessing.ScriptPostprocessing):
return pp
else:
try:
from installer import install
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg"]:
install(pkg, no_deps=True, ignore=False)
dependencies()
import rembg
if "U2NET_HOME" not in os.environ:
from modules.paths import models_path