feat(docker): comfyui docker container

.gitignore

@@ -214,3 +214,8 @@ __marimo__/
 
 # Streamlit
 .streamlit/secrets.toml
+
+# Hugging Face artifacts
+*.safetensors
+*.vae
+*.pt

Dockerfile

@@ -0,0 +1,15 @@
+FROM pytorch/pytorch:2.9.0-cuda13.0-cudnn9-runtime AS builder
+WORKDIR /app
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends git && \
+    rm -rf /var/lib/apt/lists/*
+RUN git clone --branch v0.3.67 https://github.com/comfyanonymous/ComfyUI.git ComfyUI
+WORKDIR /app/ComfyUI
+
+FROM pytorch/pytorch:2.9.0-cuda13.0-cudnn9-runtime
+WORKDIR /app
+COPY --from=builder /app/ComfyUI /app/ComfyUI
+WORKDIR /app/ComfyUI
+RUN pip install --no-cache-dir -r requirements.txt
+EXPOSE 8188
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "8188"]

compose.yaml

@@ -0,0 +1,20 @@
+services:
+  comfyui:
+    build: .
+    container_name: comfyui
+    runtime: nvidia
+    environment:
+      - CUDA_VISIBLE_DEVICES=0
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+    volumes:
+      - ./docker/models/checkpoints:/app/ComfyUI/models/checkpoints
+      - ./docker/models/loras:/app/ComfyUI/models/loras
+      - ./docker/models/vae:/app/ComfyUI/models/vae
+      - ./docker/output:/app/ComfyUI/output
+      - ./docker/workflows:/app/ComfyUI/user/default/workflows
+    ports:
+      - "8188:8188"
+    restart: unless-stopped
+    stdin_open: true
+    tty: true