diffusers_stable_diffusion_text_to_image_generation.py

1import torch
2from diffusers import DiffusionPipeline
3
4# Load the pipeline
5pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
6pipe.to("cuda")
7
8# Run the pipeline to generate an image
9prompt = "An image of a squirrel in Picasso style"
10image = pipe(prompt).images[0]
11
12# Save the image
13image.save("picasso_squirrel.png")