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

这篇文章介绍了在 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

一个文生视频的样例：

参考资料：