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

Merge pull request #13 from Thireus/patch-1

Browse Source 
Fixes #10
This commit is contained in:
AUTOMATIC1111
2023-05-17 21:52:18 +03:00
committed by GitHub
parent 64821f0476 7970e27901
commit 657ae9f548
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(