文章目录
下载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行
将下列代码注释掉并替换赋值
就不会触发安全检测了
注意公共场合小心查看图片。。