环境
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图片