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
+3 -2
View File
@@ -1,6 +1,6 @@
# Change Log for SD.Next
## Update for 2026-07-20
## Update for 2026-07-21
- **Compute**
- torch: update to `2.13.0` for CUDA, ROCm, IPEX
@@ -23,7 +23,8 @@
- lora: support diffusers trainer
- load: flux1 t5
- preview: cache image for reuse
- kanvas: fix paint combined with zoom
- kanvas: paint combined with zoom
- rembg: numba dependencies
## Update for 2026-07-14
+1 -1
View File
@@ -264,7 +264,7 @@ def run_segment(input_image: gr.Image, input_mask: np.ndarray):
def run_rembg(input_image: Image.Image, input_mask: np.ndarray):
try:
from installer import install
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg"]:
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg", "numba"]:
install(pkg, no_deps=True, ignore=False)
import rembg
except Exception as e:
+7 -3
View File
@@ -1,6 +1,12 @@
from fastapi import Body
def dependencies():
from installer import install
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg", "numba"]:
install(pkg, no_deps=True, ignore=False)
async def post_rembg(
input_image: str = Body("", title='rembg input image'),
model: str = Body("u2net", title='rembg model'),
@@ -23,9 +29,7 @@ async def post_rembg(
from modules.rembg import ben2
image = ben2.remove(input_image, refine=refine)
else:
from installer import install
for pkg in ["dctorch==0.1.2", "pymatting", "pooch", "rembg"]:
install(pkg, no_deps=True, ignore=False)
dependencies()
import rembg
image = rembg.remove( # pylint: disable=c-extension-no-member
input_image,
+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