Unverified Commit 62b3efe3 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Fix SD example typo

parent 21ceda3f
...@@ -56,7 +56,7 @@ pipe = StableDiffusionPipeline.from_pretrained( ...@@ -56,7 +56,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-3", "CompVis/stable-diffusion-v1-3",
scheduler=lms, scheduler=lms,
use_auth_token=True use_auth_token=True
) .to("cuda") ).to("cuda")
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"):
......
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