From 0b607e72a3d66ad6ef5888a1d4bd6a1493efa103 Mon Sep 17 00:00:00 2001
From: fapoverflow <50244958+fapoverflow@users.noreply.github.com>
Date: Tue, 13 Jan 2026 04:01:26 +0100
Subject: [PATCH] Update comfyui image

---
 services/comfy/Dockerfile               |  4 ++-
 services/swarmui/Dockerfile             | 36 ++++++++++++++++---------
 services/swarmui/comfy-install-linux.sh |  9 ++++---
 3 files changed, 33 insertions(+), 16 deletions(-)

diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile
index ec4dcc2..3de8686 100644
--- a/services/comfy/Dockerfile
+++ b/services/comfy/Dockerfile
@@ -178,7 +178,7 @@ FROM dev
 #FROM pytorch/pytorch:${PYTORCH_VERSION}
 
 # Defines the versions of ComfyUI, ComfyUI Manager, and PyTorch to use
-ARG COMFYUI_VERSION=v0.5.1
+ARG COMFYUI_VERSION=v0.8.2
 #ARG COMFYUI_MANAGER_VERSION=3.35
 # number of CPU's use for compilation
 ARG CPUS=10
@@ -300,6 +300,8 @@ RUN mkdir ~/nv && cd ~/nv && \
 RUN cd ~/nv && \
   git clone https://git.ffmpeg.org/ffmpeg.git ffmpeg/ && \
   cd ffmpeg && \
+  CFLAGS="-D_POSIX_C_SOURCE=200112L -D_GNU_SOURCE -I/usr/local/cuda/include" \
+  LDFLAGS="-L/usr/local/cuda/lib64" \
   ./configure \
     --enable-nonfree \
     --enable-nvenc \
diff --git a/services/swarmui/Dockerfile b/services/swarmui/Dockerfile
index f2ca9ed..f320e1e 100644
--- a/services/swarmui/Dockerfile
+++ b/services/swarmui/Dockerfile
@@ -1,6 +1,6 @@
 ARG SWARMUI_VERSION=0.9.7-Beta
-ARG COMFYUI_VERSION=v0.3.59
-ARG COMFYUI_MANAGER_VERSION=3.35
+ARG COMFYUI_VERSION=v0.8.2
+#ARG COMFYUI_MANAGER_VERSION=3.35
 ARG DOTNET_VERSION=9.0-bookworm-slim
 #ARG DOTNET_VERSION=8.0-bookworm-slim
 ARG ASPNET_VERSION=9.0-bookworm-slim
@@ -31,6 +31,9 @@ FROM mcr.microsoft.com/dotnet/aspnet:${ASPNET_VERSION}
 ARG SWARM_PATH
 ENV DOTNET_CLI_TELEMETRY_OPTOUT=1
 
+ENV USER_NAME="root"
+#ENV USER="swarmui"
+ENV GROUP_NAME="root"
 ARG SWARMUI_USER_ID=1000
 ARG SWARMUI_GROUP_ID=1000
 
@@ -38,13 +41,15 @@ ARG GPU_TYPE="nv"
 ENV NVIDIA_VISIBLE_DEVICES=all
 ENV CLI_ARGS=""
 
-RUN addgroup --gid $SWARMUI_GROUP_ID swarmui && \
-    adduser --uid $SWARMUI_USER_ID --gid $SWARMUI_GROUP_ID --gecos "" --disabled-password swarmui
+#RUN addgroup --gid $SWARMUI_GROUP_ID swarmui && \
+#    adduser --uid $SWARMUI_USER_ID --gid $SWARMUI_GROUP_ID --gecos "" --disabled-password swarmui
 
 COPY --from=build ${SWARM_PATH} "${SWARM_PATH}/"
 
-RUN mkdir -p "${SWARM_PATH}/Data" && \
-    chown -R swarmui:swarmui ${SWARM_PATH}
+RUN mkdir -p "${SWARM_PATH}/Data"
+RUN chown -R $(id -u):$(id -g) ${SWARM_PATH}
+#RUN chown -R ${USER_NAME}:${GROUP_NAME} ${SWARM_PATH}
+#RUN chown -R swarmui:swarmui ${SWARM_PATH}
 ENV HOME=${SWARM_PATH}
 
 RUN --mount=type=cache,target=/var/cache/apt \
@@ -68,15 +73,18 @@ RUN git clone --depth=1 https://github.com/comfyanonymous/ComfyUI.git /opt/comfy
     git checkout FETCH_HEAD
 
 RUN git clone --depth=1 https://github.com/Comfy-Org/ComfyUI-Manager.git /opt/comfyui-manager && \
-    cd /opt/comfyui-manager && \
-    git fetch origin ${COMFYUI_MANAGER_VERSION} && \
-    git checkout FETCH_HEAD
+    cd /opt/comfyui-manager
+#    cd /opt/comfyui-manager && \
+#    git fetch origin ${COMFYUI_MANAGER_VERSION} && \
+#    git checkout FETCH_HEAD
 
 WORKDIR ${SWARM_PATH}
-RUN chown -R swarmui:swarmui /opt/comfyui
+RUN chown -R $(id -u):$(id -g) /opt/comfyui
+#RUN chown -R ${USER_NAME}:${GROUP_NAME} /opt/comfyui
 RUN git config --global --add safe.directory /opt/comfyui
 
 ENV COMFYUI_PATH="/SwarmUI/dlbackend/ComfyUI"
+#ENV COMFYUI_PATH="/SwarmUI/dlbackend/ComfyUI"
 ENV CUSTOM_NODES_PATH="/SwarmUI/dlbackend/ComfyUI/custom_nodes"
 
 # Pre-install previously used custom nodes requirements from volume
@@ -85,10 +93,14 @@ COPY comfy-install-linux.sh ${SWARM_PATH}/launchtools/
 RUN chmod +x ${SWARM_PATH}/launchtools/comfy-install-linux.sh && \
     ${SWARM_PATH}/launchtools/comfy-install-linux.sh ${GPU_TYPE}
 
-RUN chown -R swarmui:swarmui ${COMFYUI_PATH}/venv # Reapplied again bc of permissions issues, maybe related to a symlink/docker/windows bug
+RUN chown -R $(id -u):$(id -g) ${SWARM_PATH}
+#RUN chown -R ${USER_NAME}:${GROUP_NAME} /opt/comfyui
+#RUN chown -R swarmui:swarmui ${SWARM_PATH} # Reapplied again bc of permissions issues, maybe related to a symlink/docker/windows bug
+#RUN chown -R swarmui:swarmui ${COMFYUI_PATH} # Reapplied again bc of permissions issues, maybe related to a symlink/docker/windows bug
+#RUN chown -R swarmui:swarmui ${COMFYUI_PATH}/venv # Reapplied again bc of permissions issues, maybe related to a symlink/docker/windows bug
 ENV PATH="${COMFYUI_PATH}/venv/bin:$PATH"
 ENV PYTHONPATH="${CUSTOM_NODES_PATH}:\${PYTHONPATH}"
-USER swarmui
+#USER swarmui
 EXPOSE 7801
 
 COPY entrypoint.sh /
diff --git a/services/swarmui/comfy-install-linux.sh b/services/swarmui/comfy-install-linux.sh
index 4e667ae..5760cd1 100644
--- a/services/swarmui/comfy-install-linux.sh
+++ b/services/swarmui/comfy-install-linux.sh
@@ -82,11 +82,14 @@ fi
 
 echo "install general requirements..."
 $python -s -m pip install --no-cache-dir \
-    triton \
-    sageattention \
     opencv-python \
     diffusers \
-    psutil
+    triton \
+    torchsde \
+    nvidia-ml-py \
+    sageattention \
+    psutil \
+    nvitop
 
 # Installs the required Python packages for both ComfyUI and the ComfyUI Manager
 $python -s -m pip install --no-cache-dir \