AUTOMATIC1111/stable-diffusion-webui-rembg
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21 from catboxanon/update

Browse Source 
Bump rembg to 2.0.38, support DIS models
This commit is contained in:
AUTOMATIC1111
2023-06-04 16:35:24 +03:00
committed by GitHub
parent 132c760ac3 c9c30cf3bb
commit 3d9eedbbf0
2 changed files with 3 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 import launch
if not launch.is_installed("rembg"): if not launch.is_installed("rembg"):
launch.run_pip("install rembg==2.0.30 --no-deps", "rembg") launch.run_pip("install rembg==2.0.38 --no-deps", "rembg")
for dep in ['onnxruntime', 'pymatting', 'pooch']: for dep in ['onnxruntime', 'pymatting', 'pooch']:
if not launch.is_installed(dep): if not launch.is_installed(dep):

2
scripts/postprocessing_rembg.py
View File

@@ -11,6 +11,8 @@ models = [
"u2net_human_seg", "u2net_human_seg",
"u2net_cloth_seg", "u2net_cloth_seg",
"silueta", "silueta",
"isnet-general-use",
"isnet-anime",
] ]
class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing): class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):