diff --git a/Local Homelab/ollama/compose/compose.yml b/Local Homelab/ollama/compose/compose.yml
index dfbd486..3199cba 100644
--- a/Local Homelab/ollama/compose/compose.yml	
+++ b/Local Homelab/ollama/compose/compose.yml	
@@ -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