Skip to content

Commit 0407c3e

Browse files
authored
Fix pipeline class on README (#3345)
Update README.md
1 parent 7ce3fa0 commit 0407c3e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/community/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1274,11 +1274,11 @@ mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data
12741274
init_image = download_image(img_url).resize((512, 512))
12751275
mask_image = download_image(mask_url).resize((512, 512))
12761276
mask_image = PIL.ImageOps.invert(mask_image)
1277-
pipe = DiffusionPipeline.from_pretrained(
1277+
pipe = StableDiffusionPipeline.from_pretrained(
12781278
"CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16, custom_pipeline="stable_diffusion_repaint",
12791279
)
12801280
pipe.scheduler = RePaintScheduler.from_config(pipe.scheduler.config)
12811281
pipe = pipe.to("cuda")
12821282
prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
12831283
image = pipe(prompt=prompt, image=init_image, mask_image=mask_image).images[0]
1284-
```
1284+
```

0 commit comments

Comments
 (0)