diff --git a/Local Homelab/ollama/compose/compose.yml b/Local Homelab/ollama/compose/compose.yml index 49a2794..e16bcbc 100644 --- a/Local Homelab/ollama/compose/compose.yml +++ b/Local Homelab/ollama/compose/compose.yml @@ -56,6 +56,52 @@ services: constraints: - node.role == manager + # --- 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 + " + networks: + - ai_internal + - proxy + deploy: + replicas: 1 + placement: + constraints: + - node.labels.gpu == true + + # Hardware Access + devices: + - /dev/dri:/dev/dri + - /dev/kfd:/dev/kfd + cap_add: + - IPC_LOCK + - SYS_PTRACE + group_add: + - video + - render + + volumes: + # --- UPDATED PATH --- + - /mnt/local-ssd/docker/comfyui:/workspace/ComfyUI + # System Hardware Maps + - /sys/class/drm:/sys/class/drm:ro + - /sys/class/kfd:/sys/class/kfd:ro + - /sys/devices:/sys/devices:ro + + environment: + - HSA_OVERRIDE_GFX_VERSION=10.3.0 + - CLI_ARGS=--listen 0.0.0.0 --port 8188 + networks: ai_internal: external: true