feat: provide access to ComfyUI via web

blib-la · Mar 19, 2024 · 201f64e · 201f64e
1 parent e4276de
commit 201f64e
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/start.sh b/src/start.sh
@@ -4,14 +4,17 @@
 TCMALLOC="$(ldconfig -p | grep -Po "libtcmalloc.so.\d" | head -n 1)"
 export LD_PRELOAD="${TCMALLOC}"
 
-echo "runpod-worker-comfy: Starting ComfyUI"
-python3 /comfyui/main.py --disable-auto-launch --disable-metadata &
-
-echo "runpod-worker-comfy: Starting RunPod Handler"
-
 # Serve the API and don't shutdown the container
 if [ "$SERVE_API_LOCALLY" == "true" ]; then
+    echo "runpod-worker-comfy: Starting ComfyUI"
+    python3 /comfyui/main.py --disable-auto-launch --disable-metadata --listen &
+
+    echo "runpod-worker-comfy: Starting RunPod Handler"
     python3 -u /rp_handler.py --rp_serve_api --rp_api_host=0.0.0.0
 else
+    echo "runpod-worker-comfy: Starting ComfyUI"
+    python3 /comfyui/main.py --disable-auto-launch --disable-metadata &
+
+    echo "runpod-worker-comfy: Starting RunPod Handler"
     python3 -u /rp_handler.py
 fi