1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix stable-diffusion code snippet format.

This commit is contained in:
Suraj Patil
2022-08-17 14:15:00 +05:30
committed by GitHub
parent 3cd20d59d7
commit 80e0c8ba9e

View File

@@ -47,15 +47,15 @@ from torch import autocast
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
lms = LMSDiscreteScheduler(
beta_start=0.00085,
beta_end=0.012,
beta_schedule="scaled_linear"
beta_start=0.00085,
beta_end=0.012,
beta_schedule="scaled_linear"
)
pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-3-diffusers",
scheduler=lms,
use_auth_token=True
"CompVis/stable-diffusion-v1-3-diffusers",
scheduler=lms,
use_auth_token=True
)
prompt = "a photo of an astronaut riding a horse on mars"