【踩坑指南】Stable Diffusion 服务器端部署笔记

文章目录

• 下载github文件
• 配置环境
• ckpt文件权重下载
• 生成图像
• NSFW检查（瑟图过滤）

下载github文件

https://github.com/CompVis/stable-diffusion 这个网址，下载压缩包解压，也可以用git clone下载

配置环境

这一步坑最多，建议不要按他官方的下载方式

先创建一个虚拟环境叫ldm

下载environment.yaml 文件下的包

有两个不好直接pip下载的时CLIP和taming-pytorch

可以用pip+git的方式下载:

pip install git+https://github.com/openai/CLIP.git

pip install git+https://github.com/CompVis/taming-transformers.git

其他有啥漏掉的包基本直接pip就行。

ckpt文件权重下载

下载地址：https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

随便选一个就行，下载了之后放到stable-diffusion 文件夹中的./models/ldm/stable-diffusion-v1/ 路径下。当然也可以下到其他位置，启动时用--ckpt 关键字+ 文件路径即可。

生成图像

terminal中输如python scripts/txt2img.py --prompt "white bunny, cyberpunk, rusty, by Van Gogh" 即可生成带有梵高风格的兔子图片，效果还不错

NSFW检查（瑟图过滤）

文件中自带瑟图检测，如果触发了会生成一张全黑图片或者是其他和谐图替代。其实在代码中可以注释掉。
在’scripts/txt2img.py’ 文件中 第319行
将下列代码注释掉并替换赋值

就不会触发安全检测了

注意公共场合小心查看图片。。