Madiator2011/madiator-docker-runpod
1
0
Fork 0
mirror of https://github.com/kodxana/madiator-docker-runpod.git synced 2024-11-22 02:40:12 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Comfy UI - Add ability to pass custom arguments on start

Browse source
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
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
official-templates/better-comfyui/pre_start.sh
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 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 "Starting ComfyUI setup..."
print_feedback "Syncing virtual environment..." print_feedback "Syncing virtual environment..."
@ -36,4 +44,10 @@ cd /workspace/ComfyUI
print_feedback "Starting ComfyUI server..." print_feedback "Starting ComfyUI server..."
print_feedback "ComfyUI will be available at http://0.0.0.0:3000" 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