File size: 253 Bytes
8b65f01
ca18a73
4ef68ec
1d08e64
ca18a73
 
1
2
3
4
5
6
from diffusers import DiffusionPipeline
import torch

pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float32)
pipeline.to("cpu")
pipeline("An image of a squirrel in Picasso style").images[0]