AUTOMATIC1111/stable-diffusion-webui-rembg
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg.git synced 2026-01-26 11:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

upgrade rembg to be able to use isnet-general-use

Browse Source
This commit is contained in:
Henri Huotari
2023-05-15 11:05:30 +03:00
committed by GitHub
parent 64821f0476
commit c87f5a2c46
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
install.py
View File

@@ -1,7 +1,7 @@
import launch
if not launch.is_installed("rembg"):
launch.run_pip("install rembg==2.0.30 --no-deps", "rembg")
launch.run_pip("install rembg==2.0.36 --no-deps", "rembg")
for dep in ['onnxruntime', 'pymatting', 'pooch']:
if not launch.is_installed(dep):

1
scripts/postprocessing_rembg.py
View File

@@ -6,6 +6,7 @@ import rembg
models = [
"None",
"isnet-general-use",
"u2net",
"u2netp",
"u2net_human_seg",