refactor(comfyui): use specialized pre-built ROCm image
Replaced the generic ROCm PyTorch base image and manual installation script with the yanwk/comfyui-boot:rocm image. This change simplifies the Docker Compose configuration by removing the manual git clone and dependency installation steps, leading to faster and more reliable container startups.
This commit is contained in:
@@ -58,18 +58,7 @@ services:
|
||||
|
||||
# --- IMAGE ENGINE: ComfyUI (AMD Optimized) ---
|
||||
comfyui:
|
||||
image: rocm/pytorch:rocm6.0_ubuntu22.04_py3.10_pytorch_2.1.2
|
||||
# Command to auto-install ComfyUI on first boot
|
||||
command: >
|
||||
bash -c "
|
||||
if [ ! -d /workspace/ComfyUI ]; then
|
||||
git clone https://github.com/comfyanonymous/ComfyUI /workspace/ComfyUI;
|
||||
cd /workspace/ComfyUI;
|
||||
pip install -r requirements.txt;
|
||||
fi;
|
||||
cd /workspace/ComfyUI;
|
||||
python3 main.py --listen 0.0.0.0 --port 8188 --preview-method auto
|
||||
"
|
||||
image: yanwk/comfyui-boot:rocm # Command to auto-install ComfyUI on first boot
|
||||
networks:
|
||||
- ai_internal
|
||||
- proxy
|
||||
|
||||
Reference in New Issue
Block a user