| Usage: | |
| ```python | |
| from diffusers import StableDiffusionInpaintPipeline | |
| import torch | |
| from diffusers.utils import load_image, make_image_grid | |
| import PIL | |
| # 指定模型文件路径 | |
| model_path = "Liangyingping/L2M-Inpainting" # 替换为你自己的模型路径 | |
| # 加载模型 | |
| pipe = StableDiffusionInpaintPipeline.from_pretrained( | |
| model_path, torch_dtype=torch.float16 | |
| ) | |
| pipe.to("cuda") # 如果有 GPU，可以将模型加载到 GPU 上 | |
| init_image = load_image("assets/debug_masked_image.png") | |
| mask_image = load_image("assets/debug_mask.png") | |
| W, H = init_image.size | |
| prompt = "a photo of a person" | |
| image = pipe( | |
| prompt=prompt, | |
| image=init_image, | |
| mask_image=mask_image, | |
| h=512, w=512 | |
| ).images[0].resize((W, H)) | |
| print(image.size, init_image.size) | |
| image2save = make_image_grid([init_image, mask_image, image], rows=1, cols=3) | |
| image2save.save("image2save_ours.png") | |
| ``` |