Create README.md

README.md

@@ -0,0 +1,34 @@
+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, local_files_only=True
+)
+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")
+```