AUTOMATIC1111/stable-diffusion-webui-rembg
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg.git synced 2026-03-13 17:29:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #10

Browse Source
This commit is contained in:
Thireus ☠
2023-04-02 20:03:59 +01:00
committed by GitHub
parent 64821f0476
commit 7970e27901
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/postprocessing_rembg.py
View File

@@ -45,7 +45,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
}
def process(self, pp: scripts_postprocessing.PostprocessedImage, model, return_mask, alpha_matting, alpha_matting_foreground_threshold, alpha_matting_background_threshold, alpha_matting_erode_size):
if model == "None":
if not model or model == "None":
return
pp.image = rembg.remove(