Skip to content

GitLab

"...composable_kernel_onnx.git" did not exist on "18ffbd680273c5970ff1d105c1ee6fca99e0df88"
Unverified Commit b00382e2 authored by Abdullah Alfaraj's avatar Abdullah Alfaraj Committed by GitHub
Browse files

fix docs: change sample to images (#613) 

the result of running the pipeline is stored in StableDiffusionPipelineOutput.images
parent 8b0be935
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/diffusers/pipelines/stable_diffusion/README.md
View file @ b00382e2
...@@ -67,7 +67,7 @@ pipe = pipe.to("cuda") ...@@ -67,7 +67,7 @@ pipe = pipe.to("cuda")
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"): with autocast("cuda"):
image = pipe(prompt).sample[0] image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png") image.save("astronaut_rides_horse.png")
``` ```
...@@ -89,7 +89,7 @@ pipe = StableDiffusionPipeline.from_pretrained( ...@@ -89,7 +89,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"): with autocast("cuda"):
image = pipe(prompt).sample[0] image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png") image.save("astronaut_rides_horse.png")
``` ```
...@@ -115,7 +115,7 @@ pipe = StableDiffusionPipeline.from_pretrained( ...@@ -115,7 +115,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"): with autocast("cuda"):
image = pipe(prompt).sample[0] image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png") image.save("astronaut_rides_horse.png")
``` ```
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment