Views
No views yet
1pip install accelerate transformers
2pip install git+https://github.com/huggingface/diffusers1import torch
2import torch
3from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
4from diffusers.utils import load_image
5
6image_guidance_scale = 1.5
7guidance_scale = 7.0
8model_id = "MudeHui/HQ-Edit"
9pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
10pipe.to("cuda")
11pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
12resolution = 512
13image = load_image(
14 "https://hf.co/datasets/diffusers/diffusers-images-docs/resolve/main/mountain.png"
15).resize((resolution, resolution))
16
17edit_instruction = "Turn sky into a cloudy one"
18edited_image = pipe(
19 prompt=edit_instruction,
20 image=image,
21 height=resolution,
22 width=resolution,
23 guidance_scale=image_guidance_scale,
24 image_guidance_scale=image_guidance_scale,
25 num_inference_steps=30,
26).images[0]
27edited_image.save("edited_image.png")