Erasedraw

🚀 EraseDraw模型

EraseDraw是用於圖像到圖像轉換的模型，可實現對象插入等功能，為圖像編輯帶來了新的解決方案。

🚀 快速開始

要使用這個模型，目前需要從主分支安裝diffusers庫。其API與InstructPix2Pix相同。

📦 安裝指南

使用以下命令安裝所需的庫：

pip install diffusers accelerate safetensors transformers

💻 使用示例

基礎用法

import PIL
import requests
import torch
from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler

model_id = "alpercanberk/erasedraw"
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
pipe.to("cuda")
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)

url = "https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg"
def download_image(url):
    image = PIL.Image.open(requests.get(url, stream=True).raw)
    image = PIL.ImageOps.exif_transpose(image)
    image = image.convert("RGB")
    return image
image = download_image(url)

prompt = "add sunglasses"
images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
images[0]

📄 許可證

本項目採用MIT許可證。

模型相關圖片

後續信息

代碼和數據即將發佈到GitHub（可在官網找到鏈接）。