mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-09 05:40:49 +08:00
26 lines
1017 B
Bash
Executable File
26 lines
1017 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Use libtcmalloc for better memory management
|
|
TCMALLOC="$(ldconfig -p | grep -Po "libtcmalloc.so.\d" | head -n 1)"
|
|
export LD_PRELOAD="${TCMALLOC}"
|
|
|
|
# Ensure ComfyUI-Manager runs in offline network mode inside the container
|
|
comfy-manager-set-mode offline || echo "worker-comfyui - Could not set ComfyUI-Manager network_mode" >&2
|
|
|
|
echo "worker-comfyui: Starting ComfyUI"
|
|
|
|
# Allow operators to tweak verbosity; default is DEBUG.
|
|
: "${COMFY_LOG_LEVEL:=DEBUG}"
|
|
|
|
# Serve the API and don't shutdown the container
|
|
if [ "$SERVE_API_LOCALLY" == "true" ]; then
|
|
python -u /comfyui/main.py --disable-auto-launch --disable-metadata --listen --verbose "${COMFY_LOG_LEVEL}" --log-stdout &
|
|
|
|
echo "worker-comfyui: Starting RunPod Handler"
|
|
python -u /handler.py --rp_serve_api --rp_api_host=0.0.0.0
|
|
else
|
|
python -u /comfyui/main.py --disable-auto-launch --disable-metadata --verbose "${COMFY_LOG_LEVEL}" --log-stdout &
|
|
|
|
echo "worker-comfyui: Starting RunPod Handler"
|
|
python -u /handler.py
|
|
fi |