CompVis
/

stable-diffusion-v1-2

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

patrickvonplaten commited on Dec 19, 2022

Commit

19fc41b

·

1 Parent(s): 6cf1aef

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -66,12 +66,12 @@ model_id = "CompVis/stable-diffusion-v1-2"
 device = "cuda"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
 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")
 ```

 device = "cuda"
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
 pipe = pipe.to(device)
 prompt = "a photo of an astronaut riding a horse on mars"
 with autocast("cuda"):
+    image = pipe(prompt)["sample"][0]
 image.save("astronaut_rides_horse.png")
 ```