diff --git a/src/diffusers/pipelines/stable_diffusion/README.md b/src/diffusers/pipelines/stable_diffusion/README.md index 0e6cab2e11a2..3a600c5859e9 100644 --- a/src/diffusers/pipelines/stable_diffusion/README.md +++ b/src/diffusers/pipelines/stable_diffusion/README.md @@ -67,7 +67,7 @@ pipe = pipe.to("cuda") prompt = "a photo of an astronaut riding a horse on mars" with autocast("cuda"): - image = pipe(prompt).sample[0] + image = pipe(prompt).images[0] image.save("astronaut_rides_horse.png") ``` @@ -89,7 +89,7 @@ pipe = StableDiffusionPipeline.from_pretrained( prompt = "a photo of an astronaut riding a horse on mars" with autocast("cuda"): - image = pipe(prompt).sample[0] + image = pipe(prompt).images[0] image.save("astronaut_rides_horse.png") ``` @@ -115,7 +115,7 @@ pipe = StableDiffusionPipeline.from_pretrained( prompt = "a photo of an astronaut riding a horse on mars" with autocast("cuda"): - image = pipe(prompt).sample[0] + image = pipe(prompt).images[0] image.save("astronaut_rides_horse.png") ```