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

stable diffusion官方版本复现

踩了一些坑,来记录下

环境

CentOS Linux release 7.5.1804 (Core) 服务器 RTX 3090

复现流程

按照Stable Diffusion的readme下载模型权重、我下载的是stable-diffusion-v1-4 版本的

1

因为服务器没法上huggingface,所以得把权重下载到本地,但是运行链接ln -s <path/to/model.ckpt> models/ldm/stable-diffusion-v1/model.ckpt 时会报错,因此直接将权重改名放到了这个文件夹

2

好像是需要下载检测器,那段代码直接注释掉了

3

OSError: Can’t load tokenizer for ‘openai/clip-vit-large-patch14’.

还是因为被墙了,从clip上下载权重,然后修改文件夹名字为clip-vit-large-patch14 同时修改diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/modules/encoders/modules.py文件内的路径

结果

python scripts/txt2img.py --ckpt "sd-v1-4.ckpt" --prompt "a photograph of an astronaut riding a elephant" --plms
历经千辛万苦,生成了demo图片

参考资料

Stable Diffusion搭建全过程记录,生成自己的专属艺术照 stable-diffusion官方版本地复现手册(2023年4月10日)

更新时间 2024-06-13