Sign inSign up

ocpub/oc9-stable-diffusion-webui

By ocpub

Updated 4 months ago

Image
0

170

ocpub/oc9-stable-diffusion-webui repository overview

Stable Diffusion WebUI on OpenCloudOS 9

基本信息

  • 框架版本: 1.9.4
  • 基础镜像: opencloudos/opencloudos9-cuda-devel:12.8
  • Python 版本: 3.11
  • CUDA 版本: 12.8
  • 核心组件:
    • Stable Diffusion WebUI 1.9.4
    • PyTorch 2.11.0+cu128
    • Gradio 3.41.2
    • xformers 0.0.35
    • transformers 4.30.2

构建

docker build -t oc9-stable-diffusion-webui:1.9.4 .

构建时间约 15-20 分钟,会自动下载 SD 1.5 模型(4GB)。

使用示例

1. 启动容器
docker run -d \
  --name sd-webui \
  --gpus all \
  -p 7860:7860 \
  -v ~/sd-outputs:/app/outputs \
  oc9-stable-diffusion-webui:1.9.4
2. 访问 Web UI

浏览器访问:http://localhost:7860

首次启动需要 10-15 秒加载模型。

3. API 调用示例
# 文本生成图像
curl -X POST http://localhost:7860/sdapi/v1/txt2img \
  -H "Content-Type: application/json" \
  -d '{
    "prompt": "a beautiful sunset over mountains",
    "steps": 20,
    "width": 512,
    "height": 512
  }'
4. 容器内测试
# 复制测试脚本到容器
docker cp test.sh sd-webui:/app/test-inside.sh

# 在容器内执行测试
docker exec sd-webui chmod +x /app/test-inside.sh
docker exec sd-webui /app/test-inside.sh

功能特性

  • ✓ txt2img 文本生成图像
  • ✓ img2img 图像转换
  • ✓ Inpainting 图像修复
  • ✓ LoRA 模型支持
  • ✓ xformers 内存优化
  • ✓ REST API 接口
  • ✓ 多种采样器(Euler、DPM++、DDIM 等)

系统要求

最低配置
  • GPU: NVIDIA GPU with 6GB+ VRAM
  • RAM: 16GB+
  • 存储: 30GB+
推荐配置
  • GPU: NVIDIA RTX 3060 (12GB VRAM) 或更高
  • RAM: 32GB+
  • 存储: 50GB+ SSD

性能参考

在 RTX 4090 (24GB) 上测试:

  • 512x512 图像,20 步采样:约 1-2 秒
  • 启用 xformers 优化
  • 模型加载时间:约 11 秒

已知问题

  1. 首次启动较慢:需要初始化 repositories 和加载模型,约需 15-20 秒
  2. 网络依赖:构建时需要从 GitHub 和 HuggingFace 下载资源,建议使用网速快的服务器
  3. 显存要求:生成 512x512 图像至少需要 6GB 显存,推荐 8GB+

低显存优化

如果 GPU 显存不足,可以使用以下参数:

# 6-8GB 显存
docker run -d --name sd-webui --gpus all -p 7860:7860 \
  oc9-stable-diffusion-webui:1.9.4 \
  python3.11 launch.py --listen --port 7860 --medvram --xformers

# 4-6GB 显存
docker run -d --name sd-webui --gpus all -p 7860:7860 \
  oc9-stable-diffusion-webui:1.9.4 \
  python3.11 launch.py --listen --port 7860 --lowvram --xformers

添加自定义模型

# 挂载模型目录
docker run -d \
  --name sd-webui \
  --gpus all \
  -p 7860:7860 \
  -v ~/sd-models:/app/models/Stable-diffusion \
  -v ~/sd-outputs:/app/outputs \
  oc9-stable-diffusion-webui:1.9.4

.safetensors.ckpt 模型文件放入 ~/sd-models 目录即可。

参考资源

许可证

本镜像遵循 AGPL-3.0 许可证。

Tag summary

Content type

Image

Digest

sha256:6eabe8323

Size

16 GB

Last updated

4 months ago

docker pull ocpub/oc9-stable-diffusion-webui:1.9.4