Instruct Pix2pix

🚀 InstructPix2Pix：学习遵循图像编辑指令

InstructPix2Pix是一个能够学习遵循图像编辑指令的项目，可实现图像到图像的转换，为图像编辑提供了新的解决方案。

🚀 快速开始

要使用 InstructPix2Pix，目前需要使用 main 分支来安装 diffusers。该管道将在下一版本中正式可用。

📦 安装指南

pip install diffusers accelerate safetensors transformers

💻 使用示例

基础用法

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

model_id = "timbrooks/instruct-pix2pix"
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 = "turn him into cyborg"
images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
images[0]

📄 许可证

本项目采用 MIT 许可证。

GitHub链接：https://github.com/timothybrooks/instruct-pix2pix