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/