from diffusers import DiffusionPipeline
import torch

device = "cuda"
generator = torch.Generator(device).manual_seed(2500)
prompt = "An astronaut riding a green horse"

pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
pipe.to(device)

images = pipe(prompt=prompt, generator=generator).images[0]
images
