Skip to content

GitLab

Commit 89509230 authored by anton-l's avatar anton-l
Browse files

Merge remote-tracking branch 'origin/main'

parents 577a6a65 62b3efe3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 89509230
...@@ -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