1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Match params with official Stable Diffusion lib (#192)

https://github.com/CompVis/stable-diffusion
This commit is contained in:
apolinario
2022-08-16 21:52:22 +01:00
committed by GitHub
parent 4b02f53e62
commit e36a36788e

View File

@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"):
image = pipe(prompt, width=768, guidance_scale=7)["sample"][0]
image = pipe(prompt, width=512, guidance_scale=7.5)["sample"][0]
image.save("astronaut_rides_horse.png")
```