Skip to content

Commit e36a367

Browse files
authored
Match params with official Stable Diffusion lib (huggingface#192)
https://github.com/CompVis/stable-diffusion
1 parent 4b02f53 commit e36a367

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
6060

6161
prompt = "a photo of an astronaut riding a horse on mars"
6262
with autocast("cuda"):
63-
image = pipe(prompt, width=768, guidance_scale=7)["sample"][0]
63+
image = pipe(prompt, width=512, guidance_scale=7.5)["sample"][0]
6464

6565
image.save("astronaut_rides_horse.png")
6666
```

0 commit comments

Comments
 (0)