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下载模型
git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
cd ./stable-diffusion-v1-4
git lfs install # 安装git-lfs
git lfs pull
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
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如下:
© 版权声明
文章版权归作者所有,未经允许请勿转载。