diff --git a/scripts/postprocessing_rembg.py b/scripts/postprocessing_rembg.py index da80e86..50b3fec 100644 --- a/scripts/postprocessing_rembg.py +++ b/scripts/postprocessing_rembg.py @@ -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(