File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/diffusers/pipelines/stable_diffusion Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -61,7 +61,7 @@ pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
6161pipe = pipe.to(" cuda" )
6262
6363prompt = " a photo of an astronaut riding a horse on mars"
64- image = pipe(prompt).sample [0 ]
64+ image = pipe(prompt).images [0 ]
6565
6666image.save(" astronaut_rides_horse.png" )
6767```
@@ -80,7 +80,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
8080).to(" cuda" )
8181
8282prompt = " a photo of an astronaut riding a horse on mars"
83- image = pipe(prompt).sample [0 ]
83+ image = pipe(prompt).images [0 ]
8484
8585image.save(" astronaut_rides_horse.png" )
8686```
@@ -99,7 +99,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
9999).to(" cuda" )
100100
101101prompt = " a photo of an astronaut riding a horse on mars"
102- image = pipe(prompt).sample [0 ]
102+ image = pipe(prompt).images [0 ]
103103
104104image.save(" astronaut_rides_horse.png" )
105105```
You can’t perform that action at this time.
0 commit comments