CompVis
/

stable-diffusion-v1-4

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

patrickvonplaten commited on Sep 15, 2022

Commit

e175af9

·

1 Parent(s): 48a1bd8

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -79,7 +79,7 @@ pipe = pipe.to(device)
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
-    image = pipe(prompt, guidance_scale=7.5)["sample"][0]
 image.save("astronaut_rides_horse.png")
 ```
@@ -96,7 +96,7 @@ pipe = pipe.to(device)
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
-    image = pipe(prompt, guidance_scale=7.5)["sample"][0]
 image.save("astronaut_rides_horse.png")
 ```
@@ -114,7 +114,7 @@ pipe = pipe.to("cuda")
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
-    image = pipe(prompt, guidance_scale=7.5)["sample"][0]
 image.save("astronaut_rides_horse.png")
 ```

 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("astronaut_rides_horse.png")
 ```
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("astronaut_rides_horse.png")
 ```
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("astronaut_rides_horse.png")
 ```