Skip to content

Commit 480d884

Browse files
authored
[doc] update example for pix2pix (#2101)
update example for pix2pix
1 parent 9dbf78e commit 480d884

File tree

2 files changed

+7
-9
lines changed

2 files changed

+7
-9
lines changed

docs/source/en/api/pipelines/stable_diffusion/pix2pix.mdx

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,11 +45,9 @@ import torch
4545
from diffusers import StableDiffusionInstructPix2PixPipeline
4646

4747
model_id = "timbrooks/instruct-pix2pix"
48-
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
49-
model_id, torch_dtype=torch.float16, safety_checker=None
50-
).to("cuda")
48+
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
5149

52-
url = "https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg"
50+
url = "https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/mountain.png"
5351

5452

5553
def download_image(url):
@@ -61,9 +59,9 @@ def download_image(url):
6159

6260
image = download_image(url)
6361

64-
prompt = "turn him into a cyborg"
65-
images = pipe(prompt, image=image, num_inference_steps=10, guidance_scale=1.1, image_guidance_scale=1).images
66-
images[0].save("david_cyborg.png")
62+
prompt = "make the mountains snowy"
63+
edit = pipe(prompt, image=image, num_inference_steps=20, image_guidance_scale=1.5, guidance_scale=7).images[0]
64+
images[0].save("snowy_mountains.png")
6765
```
6866

6967
## StableDiffusionInstructPix2PixPipeline

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def __call__(
208208
... return PIL.Image.open(BytesIO(response.content)).convert("RGB")
209209
210210
211-
>>> img_url = "https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg"
211+
>>> img_url = "https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/mountain.png"
212212
213213
>>> image = download_image(img_url).resize((512, 512))
214214
@@ -217,7 +217,7 @@ def __call__(
217217
... )
218218
>>> pipe = pipe.to("cuda")
219219
220-
>>> prompt = "turn him into cyborg"
220+
>>> prompt = "make the mountains snowy"
221221
>>> image = pipe(prompt=prompt, image=image).images[0]
222222
```
223223

0 commit comments

Comments
 (0)