获取stable diffusion模型在CPU上运行

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

目录

一、注册hugging face账号并登录

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

二、阅读license并同意条款

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

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

三、使用git-lfs下载模型

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

四、在CPU上运行

运行前先安装diffusers：

然后运行以下python代码：

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如下：