Skip to content

本地搭建Stable diffusion

本篇blog搬运自B站UP主荒野大皮克

git clone Stable diffusion 开源项目到本地

1. 安装Stable diffusion

如果是 Mac/Linux 系统项目根目录下运行文件 ./webui.sh,如果是Windows系统在项目根目录下运行文件 ./webui.bat

脚本运行过程中可能会失败,基本都是模型或者开源项目clone失败,基本根据报错信息中的[项目/资源]地址手动下载并且放入对应目录即可

2. Stable Diffusion运行效果

Stable Diffusion运行效果

3. 安装ollama

本地安装ollama

4. 下载大模型

在 ollama 模型库中选择 多模态模型 llAVA 用来做图片识别

shell
ollama run llava

通过ollama 下载 Stable Diffusion提示词模型 stable-diffusion-prompt-generator

shell
ollama run brxce/stable-diffusion-prompt-generator

5. 安装open-webui

shell
docker run -d -p 18800:8080 --add-host=host.docker.internal:host-gateway -v open-webui:/app/backend/data --name open-webui --restart always ghcr.io/open-webui/open-webui:main

启动之后通过本地链接 http://localhost:18800 进入 open-webui ,open-webui是开放注册的,自己注册一个账号进入界面.

open-webui 界面

open-webui

设置本地 Stable Diffusion模型
  1. 在Stable Diffusion本地项目中,使用 ./webui.sh --api 启动模型,切记加入 --api 参数,否则无法使用!
  2. 填入本地启动的 Stable Diffusion 模型地址 ,我的是 http://127.0.0.1:7860/ 但是因为 我们的open-webui是在docker容器中启动的,所以要将地址换成: http://host.docker.internal:7860/

设置本地 Stable Diffusion模型

6. 测试图片识别

测试llava多模态模型

7. 测试通过提示词模型生成图片

测试通过提示词模型生成图片测试通过提示词模型生成图片-success