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:
2025-12-24 21:23:06 -05:00
parent 31e7a30681
commit 89ad1f134e

View File

@@ -58,18 +58,7 @@ services:
# --- IMAGE ENGINE: ComfyUI (AMD Optimized) --- # --- IMAGE ENGINE: ComfyUI (AMD Optimized) ---
comfyui: comfyui:
image: rocm/pytorch:rocm6.0_ubuntu22.04_py3.10_pytorch_2.1.2 image: yanwk/comfyui-boot:rocm # Command to auto-install ComfyUI on first boot
# 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
"
networks: networks:
- ai_internal - ai_internal
- proxy - proxy