stable_diffusion_2.py

# pip install diffusers transformers accelerate

# Load model
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
import torch

repo_id = "stabilityai/stable-diffusion-2"
device = "cuda"

pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.float16, revision="fp16")
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
pipe = pipe.to(device)

# Generate your image
prompt = "High quality photo of an astronaut riding a horse in space"
pipe(prompt, guidance_scale=9).images[0]