Add swarmui

services/convert2safetensors/entrypoint.sh

@@ -2,5 +2,4 @@
 set -ex
 
 echo "Running container as $USER"
-echo "Running container as $USER..."
 python convert_2_safetensors.py --source ./models/input --dest ./models/output --allow-pickle

services/swarmui/Dockerfile

@@ -0,0 +1,68 @@
+ARG SWARMUI_VERSION=0.9.6-Beta
+ARG DOTNET_VERSION=8.0-bookworm-slim
+
+FROM mcr.microsoft.com/dotnet/sdk:${DOTNET_VERSION} AS build
+ENV DOTNET_CLI_TELEMETRY_OPTOUT=1
+ENV SWARM_PATH="/SwarmUI"
+#ENV SWARM_PATH=/opt/swarmui
+
+RUN --mount=type=cache,target=/var/cache/apt \
+    apt-get update && \
+    apt-get install -y git
+
+WORKDIR ${SWARM_PATH}
+
+# Clones the SwarmUI repository and checks out the latest release
+RUN git clone --depth=1 https://github.com/mcmonkeyprojects/SwarmUI.git ${SWARM_PATH} && \
+    cd ${SWARM_PATH} && \
+    git fetch origin ${SWARMUI_VERSION} && \
+    git checkout FETCH_HEAD
+
+RUN dotnet build src/SwarmUI.csproj --configuration Release -o ./bin
+
+# RUN
+FROM mcr.microsoft.com/dotnet/aspnet:8.0-bookworm-slim
+ENV DOTNET_CLI_TELEMETRY_OPTOUT=1
+ENV SWARM_PATH="/SwarmUI"
+#ENV SWARM_PATH=/opt/swarmui
+ARG SWARMUI_USER_ID=1000
+ARG SWARMUI_GROUP_ID=1000
+
+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
+
+COPY --from=build ${SWARM_PATH} "${SWARM_PATH}/"
+
+RUN mkdir -p "${SWARM_PATH}/Data" && \
+    chown -R swarmui:swarmui ${SWARM_PATH}
+ENV HOME=${SWARM_PATH}
+
+RUN --mount=type=cache,target=/var/cache/apt \
+    apt update --assume-yes && \
+    apt install -y \
+        git \
+        wget \
+        build-essential \
+        python3.11 \
+        python3.11-venv \
+        python3.11-dev \
+        python3-pip \
+        ffmpeg  \
+        libglib2.0-0 \
+        libgl1
+
+WORKDIR ${SWARM_PATH}
+
+USER swarmui
+
+RUN chmod +x ${SWARM_PATH}/launchtools/comfy-install-linux.sh && \
+    ${SWARM_PATH}/launchtools/comfy-install-linux.sh ${GPU_TYPE}
+
+EXPOSE 7801
+
+COPY entrypoint.sh /
+ENTRYPOINT ["/entrypoint.sh", "--launch_mode", "none", "--host", "0.0.0.0"]

services/swarmui/entrypoint.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+
+# Ensure correct local path.
+SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+cd "$SCRIPT_DIR"
+
+# Add dotnet non-admin-install to path
+export PATH="$SCRIPT_DIR/.dotnet:~/.dotnet:$PATH"
+
+# Default env configuration, gets overwritten by the C# code's settings handler
+export ASPNETCORE_ENVIRONMENT="Production"
+export ASPNETCORE_URLS="http://*:7801"
+# Actual runner.
+cd "$HOME"
+dotnet ./bin/SwarmUI.dll "$@"
+
+# Exit code 42 means restart, anything else = don't.
+if [ $? == 42 ]; then
+    . /entrypoint.sh "$@"
+fi