获取stable diffusion模型在CPU上运行

新闻 2023-04-20
4311

Stable Diffusion v1-4 Model Card:CompVis/stable-diffusion-v1-4 · Hugging Face

目录

一、注册hugging face账号并登录

为了获取stable diffusion模型,首先需要有hugging face账号。如果没有hugging face账号,可以通过上面那个页面的右上角通过邮箱注册,注册后打开邮箱confirm账号就生效了。有了账号后,通过上面那个页面的右上角登录账号。

二、阅读license并同意条款

登录账号页面刷新后,会出现一个license的条款:

获取stable diffusion模型在CPU上运行

阅读条款后,如果你同意条款,可以勾选倒数第二行的方框,然后按下面一行的Access repository按钮。

三、使用git-lfs下载模型

git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
cd ./stable-diffusion-v1-4
git lfs install # 安装git-lfs
git lfs pull

完成以上三个步骤,stable-diffusion-v1-4模型就下载下来了。

四、在CPU上运行

运行前先安装diffusers:

pip install --upgrade diffusers transformers scipy

然后运行以下python代码:

from torch import autocast
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler

pipe = StableDiffusionPipeline.from_pretrained("../../stable-diffusion-v1-4")

prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]

image.save("astronaut_rides_horse.png")

上面的代码中使用的text是"a photo of an astronaut riding a horse on mars" (一张宇航员在火星上骑马的照片)

生成的image如下:

版权声明: 发表于 2023-04-20。
转载请注明:获取stable diffusion模型在CPU上运行 | 非常AI

相关文章