Skip to main content

运行 stable diffusion

info

stable diffusion 是一种潜在的文本到图像扩散模型,能够在输入任何文本的情况下生成图像, 也可以由图片生成图片

准备环境

  • python
  • pytorch

pytorch 可以提前下载 whl 包安装,比直接 pip install 快。 pytorch whl下载地址

执行 pip install + wheel文件名

# 举例
pip install torch-2.0.0-cu118-cp310-cp310-win_amd64.whl

启动时,会把所有所有依赖都存在 stable-diffusion-webui/venv 目录下

代码下载

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

模型下载

以下有几个常用的知名模型下载网站,如果有需要的可以先去下载自己喜欢的模型再来启动 stable diffusion 这样可以避免等待模型下载时间过长, 也可以之后再来下载需要的模型

运行 stable diffusion

进入代码文件夹 stable-diffusion-webui

  • Windows 双击 webui-user.bat 运行
  • Linux / MacOs 执行 webui-user.sh 运行

初始启动会下载默认的SD模型,模型会比较大(4G), 如果有需要的模型,也可以提前下载,放入 stable diffusion/models 目录

启动后日志不报错即可,可以在日志中看到访问URL http://127.0.0.1:7860

打开链接后可以看到基于 gradio 部署的AI算法可视化界面 stable-diffusion.png