Unverified Commit cd4e053e authored by catboxanon's avatar catboxanon Committed by GitHub
Browse files

Simply img2img mask conversion, fix threshold

parent 2dc2bc4a
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -129,7 +129,7 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s
        mask = None
        mask = None
    elif mode == 2:  # inpaint
    elif mode == 2:  # inpaint
        image, mask = init_img_with_mask["image"], init_img_with_mask["mask"]
        image, mask = init_img_with_mask["image"], init_img_with_mask["mask"]
        mask = mask.convert("RGBA").split()[3].convert("L").point(lambda x: 255 if x > 0 else 0)
        mask = mask.split()[-1].convert('L').point(lambda x: 255 if x > 128 else 0)
        image = image.convert("RGB")
        image = image.convert("RGB")
    elif mode == 3:  # inpaint sketch
    elif mode == 3:  # inpaint sketch
        image = inpaint_color_sketch
        image = inpaint_color_sketch