Skip to content

GitLab

Unverified Commit 80e0c8ba authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

fix stable-diffusion code snippet format.

parent 3cd20d59
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
README.md
View file @ 80e0c8ba
...@@ -47,15 +47,15 @@ from torch import autocast ...@@ -47,15 +47,15 @@ from torch import autocast
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
lms = LMSDiscreteScheduler( lms = LMSDiscreteScheduler(
beta_start=0.00085, beta_start=0.00085,
beta_end=0.012, beta_end=0.012,
beta_schedule="scaled_linear" beta_schedule="scaled_linear"
) )
pipe = StableDiffusionPipeline.from_pretrained( pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-3-diffusers", "CompVis/stable-diffusion-v1-3-diffusers",
scheduler=lms, scheduler=lms,
use_auth_token=True use_auth_token=True
) )
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
......
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