runwayml
/

stable-diffusion-v1-5

patrickvonplaten commited on Oct 20, 2022

Commit

245f52e

•

2 Parent(s): 7621c1d 98459e8

Merge branch 'main' of https://huggingface.co/runwayml/stable-diffusion-v1-5 into main

Files changed (1) hide show

README.md CHANGED Viewed

@@ -32,11 +32,12 @@ You can use this both with the [🧨Diffusers library](https://github.com/huggin
 ```py
 from diffusers import StableDiffusionPipeline
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
 pipe = pipe.to(device)
 prompt = "a photo of an astronaut riding a horse on mars"
-image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("astronaut_rides_horse.png")
 ```

 ```py
 from diffusers import StableDiffusionPipeline
+model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
 pipe = pipe.to(device)
 prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]
 image.save("astronaut_rides_horse.png")
 ```