mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
fix rembg dependencies
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
+3
-2
@@ -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
@@ -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:
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user