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_serviceprofiles: ["comfy"]build: ./services/comfy/image: sd-comfy:7environment:- 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/