当前位置:AIGC资讯 > AIGC > 正文

【踩坑指南】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行
将下列代码注释掉并替换赋值

就不会触发安全检测了

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

更新时间 2023-12-20