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 @@ -67,7 +67,7 @@ pipe = pipe.to("cuda")
6767
6868prompt = " a photo of an astronaut riding a horse on mars"
6969with autocast(" cuda" ):
70- image = pipe(prompt).sample [0 ]
70+ image = pipe(prompt).images [0 ]
7171
7272image.save(" astronaut_rides_horse.png" )
7373```
@@ -89,7 +89,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
8989
9090prompt = " a photo of an astronaut riding a horse on mars"
9191with autocast(" cuda" ):
92- image = pipe(prompt).sample [0 ]
92+ image = pipe(prompt).images [0 ]
9393
9494image.save(" astronaut_rides_horse.png" )
9595```
@@ -115,7 +115,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
115115
116116prompt = " a photo of an astronaut riding a horse on mars"
117117with autocast(" cuda" ):
118- image = pipe(prompt).sample [0 ]
118+ image = pipe(prompt).images [0 ]
119119
120120image.save(" astronaut_rides_horse.png" )
121121```
You can’t perform that action at this time.
0 commit comments