File tree 1 file changed +3
-3
lines changed
src/diffusers/pipelines/stable_diffusion
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")
61
61
pipe = pipe.to(" cuda" )
62
62
63
63
prompt = " a photo of an astronaut riding a horse on mars"
64
- image = pipe(prompt).sample [0 ]
64
+ image = pipe(prompt).images [0 ]
65
65
66
66
image.save(" astronaut_rides_horse.png" )
67
67
```
@@ -80,7 +80,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
80
80
).to(" cuda" )
81
81
82
82
prompt = " a photo of an astronaut riding a horse on mars"
83
- image = pipe(prompt).sample [0 ]
83
+ image = pipe(prompt).images [0 ]
84
84
85
85
image.save(" astronaut_rides_horse.png" )
86
86
```
@@ -99,7 +99,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
99
99
).to(" cuda" )
100
100
101
101
prompt = " a photo of an astronaut riding a horse on mars"
102
- image = pipe(prompt).sample [0 ]
102
+ image = pipe(prompt).images [0 ]
103
103
104
104
image.save(" astronaut_rides_horse.png" )
105
105
```
You can’t perform that action at this time.
0 commit comments