stable-diffusion-webui-docker 构建 comfy-ui

莫扎特不唱摇篮曲2025-03-15 18:24

Ubuntu 安装 stable-diffusion-webui-docker 常见问题处理方法这篇文章介绍了在 Ubuntu 上安装 stable-diffusion-webui-docker，运行 docker compose --profile auto up --build 构建出的界面是 stable-diffusion-webui，如果运行 docker compose --profile comfy up --build，那么就是生成的 comfy-ui 界面，如下图所示

同理，如需指定 comfy 运行的gpu，可以修改 docker-compose.yaml，如下，指定 --cuda-device 0：

复制代码

  comfy: &comfy
    <<: *base_service
    profiles: ["comfy"]
    build: ./services/comfy/
    image: sd-comfy:7
    environment:
      - CLI_ARGS=--cuda-device 0

在 /stable-diffusion-webui-docker/services/comfy/Dockerfile 文件中需要注意两点：

1、FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime pytorch 镜像需要和本地环境保持一致

2、需要删除 git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \ 这一行，直接使用最新的 comfyui 即可

如果你还用到了文生视频等功能，还需要放开一些目录映射，详见文件 stable-diffusion-webui-docker/services/comfy/extra_model_paths.yaml，如下：

我统一修改为：

复制代码

style_models: models/style_models
t2i_adapter: models/t2i_adapter
clip_vision: models/clip_vision
diffusers: models/diffusers
text_encoders: models/text_encoders
diffusion_models: models/Stable-diffusion

一个文生视频的样例：

参考资料：
https://comfyanonymous.github.io/ComfyUI_examples/wan/
https://modelscope.cn/models/Comfy-Org/Wan_2.1_ComfyUI_repackaged/