stable-diffusion AI绘画安装教程(附:stable diffusion官网地址)

共计 649 个字符,预计需要花费 2 分钟才能阅读完成。

本文介绍 stable-diffusion 本地执行方案。stable-diffusion 的原始开源 github 地址是:https://github.com/CompVis/stable-diffusion。

安装过程

安装过程需要下载 sd-v1-4.ckpt 权重文件,这是一个大小为 4GB 的文件。如果需要向主机传输,则比较麻烦,需要要求速度快。推荐租用 GPU 主机,或使用自己的主机,比较便宜的租用服务是 http://gpu.ai-galaxy.cn/console/index。选择云主机时,要求现存至少 24G。

可以跑 512*512 的图片,再大则显存不够。按照 git 里面的说明部署文件(https://github.com/CompVis/stable-diffusion),安装 python 虚拟环境,而虚拟环境请使用文末代码内内容。

stable-diffusion AI 绘画安装教程(附:stable diffusion 官网地址)

执行代码时,可能会出现执行生成文件的报错。需要将以下代码更新:

from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer

更新为

from taming.modules.vqvae.quantize import VectorQuantizer as VectorQuantizer

执行过程中需要加载各种 modal 文件,这非常漫长。执行完毕后,生成每个图片大概需要 10-15 秒,速度比 MD 要快很多。

stable diffusion 官网地址

https://stablediffusionweb.com/

stable-diffusion AI 绘画安装教程(附:stable diffusion 官网地址)

正文完