# https://github.com/huggingface/diffusers
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
from diffusers import DiffusionPipeline
import torch
import numpy as np
import matplotlib.pyplot as plt
print(torch.__version__)
print(torch.cuda.is_available())
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
pipeline.to("cuda")
image = pipeline("An image of a squirrel in Picasso style").images[0]
# 假设你的image对象是一个PIL图像
image_np = np.array(image)
plt.imshow(image_np)
plt.show()
diffusers笔记
最新推荐文章于 2024-04-29 09:09:06 发布