Stable Diffusion Pipeline

diffusers.py

# make sure you're logged in with `huggingface-cli login`
import os
from torch import autocast
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler

lms = LMSDiscreteScheduler(
    beta_start=0.00085, 
    beta_end=0.012, 
    beta_schedule="scaled_linear"
)

pipe = StableDiffusionPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-3", 
    scheduler=lms,
    use_auth_token=True,
    cache_dir=os.getenv("cache_dir", "./models")
).to("cpu")

prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cpu"):
    image = pipe(prompt)["sample"][0]  
    
image.save("astronaut_rides_horse.png")