Skip to content

Instantly share code, notes, and snippets.

@timmyers
Created May 10, 2023 20:15
Show Gist options
  • Save timmyers/70294bf37381edb89f91de4fc903c67c to your computer and use it in GitHub Desktop.
Save timmyers/70294bf37381edb89f91de4fc903c67c to your computer and use it in GitHub Desktop.
Stable Diffusion 2.1 Example
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
import torch
model_id = "stabilityai/stable-diffusion-2-1-base"
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment