Comfy UI - Add ability to pass custom arguments on start

This commit is contained in:
Madiator2011 2024-07-13 20:19:26 +02:00 committed by GitHub
parent 656abb6292
commit a833efc5d8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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
# 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