谷歌colab测试diffusers

pip install --upgrade diffusers


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

访问地址

https://colab.research.google.com/