Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "ece55227ffc60f7bb09442b390a02e45ac0438f8"
Unverified Commit e36a3678 authored by apolinario's avatar apolinario Committed by GitHub
Browse files

Match params with official Stable Diffusion lib (#192) 

https://github.com/CompVis/stable-diffusion
parent 4b02f53e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ e36a3678
...@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained( ...@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"): 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") image.save("astronaut_rides_horse.png")
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment