From a833efc5d80ac41f47a342a4c426a3a502f6e6a7 Mon Sep 17 00:00:00 2001 From: Madiator2011 Date: Sat, 13 Jul 2024 20:19:26 +0200 Subject: [PATCH] Comfy UI - Add ability to pass custom arguments on start --- official-templates/better-comfyui/pre_start.sh | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/official-templates/better-comfyui/pre_start.sh b/official-templates/better-comfyui/pre_start.sh index a56a5ac..5e0142a 100644 --- a/official-templates/better-comfyui/pre_start.sh +++ b/official-templates/better-comfyui/pre_start.sh @@ -15,6 +15,14 @@ rsync_with_progress() { stdbuf -i0 -o0 -e0 rsync -au --info=progress2 "$@" | stdbuf -i0 -o0 -e0 tr '\r' '\n' | stdbuf -i0 -o0 -e0 grep -oP '\d+%|\d+.\d+[mMgG]' | tqdm --bar-format='{l_bar}{bar}' --total=100 --unit='%' > /dev/null } +# Check for required commands +for cmd in stdbuf rsync tr grep tqdm; do + if ! command -v $cmd &> /dev/null; then + echo "$cmd could not be found, please install it." + exit 1 + fi +done + print_feedback "Starting ComfyUI setup..." print_feedback "Syncing virtual environment..." @@ -36,4 +44,10 @@ cd /workspace/ComfyUI print_feedback "Starting ComfyUI server..." print_feedback "ComfyUI will be available at http://0.0.0.0:3000" -exec /workspace/venv/bin/python main.py --listen --port 3000 \ No newline at end of file + +# Check if CUSTOM_ARGS is set and not empty +if [ -n "$CUSTOM_ARGS" ]; then + exec /workspace/venv/bin/python main.py --listen --port 3000 $CUSTOM_ARGS +else + exec /workspace/venv/bin/python main.py --listen --port 3000 +fi