INNER CODE UNIT · Python

inpaint

zhulinyv/Semi-Auto-NovelAI-to-Pixiv · src/batch_inpaint.py:128

def inpaint(img_path, mask_path, inpaint_overlay, *args, **kwargs):
    imginfo = get_img_info(img_path)
    json_for_inpaint = prepare_json(imginfo, img_path)
    json_for_inpaint["parameters"]["mask"] = img_to_base64(mask_path)
    if env.model == "nai-diffusion-4-curated-preview":
        model = "nai-diffusion-4-curated-inpainting"
    elif env.model == "nai-diffusion-4-full":
        model = "nai-diffusion-4-full-inpainting"
    else:
        model = f"{env.model}-inpainting" if env.model != "nai-diffusion-2" else "nai-diffusion-3-inpainting"
    json_for_inpaint["model"] = model
    json_for_inpaint["action"] = "infill"
    json_for_inpaint["add_original_image"] = inpaint_overlay
    json_for_inpaint["strength"] = kwargs["inpaint_strength"]
    json_for_inpaint["noise"] = kwargs["inpaint_noise"]

    if "nai-diffusion-4" in env.model:
        json_for_inpaint["parameters"]["use_coords"] = args[0]

View source record →