mdztxi / app.py
Geek7's picture
Update app.py
1d08e64 verified
raw
history blame
253 Bytes
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]