Erasedraw

🚀 EraseDraw Model

These model weights are designed for object insertion in image - to - image tasks, offering a powerful solution for modifying images.

🚀 Quick Start

These are the model weights for EraseDraw.

📦 Installation

To use this model, you need to install diffusers using the main branch for now. The API is the same as that of InstructPix2Pix.

pip install diffusers accelerate safetensors transformers

💻 Usage Examples

Basic Usage

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]

📄 License

This project is licensed under the MIT license.

Code and data are coming soon to GitHub (find link on website).