mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-16 01:37:04 +08:00
d94b0315ee
- Update repository URL from jeffy5 to ru4ls (correct repo) - Add pip install step for Nano Banana dependencies - Add CUSTOM_NODES_DIR variable to entrypoint script - Node now installs dependencies from requirements.txt automatically Repository: https://github.com/ru4ls/ComfyUI_Nano_Banana
87 lines
3.4 KiB
Bash
Executable File
87 lines
3.4 KiB
Bash
Executable File
#!/bin/bash
|
|
# Docker entrypoint script for ComfyUI
|
|
# Handles virtual environment setup, dependency installation, custom nodes, and optional OTEL instrumentation
|
|
|
|
set -e
|
|
|
|
COMFYUI_DIR="/app/ComfyUI"
|
|
VENV_DIR="/app/venv"
|
|
WORKDIR="${COMFYUI_DIR}"
|
|
CUSTOM_NODES_DIR="${COMFYUI_DIR}/custom_nodes"
|
|
|
|
cd "${WORKDIR}"
|
|
|
|
# Create virtual environment if it doesn't exist
|
|
if [ ! -f "${VENV_DIR}/bin/activate" ]; then
|
|
echo "Creating virtual environment..."
|
|
python -m venv "${VENV_DIR}"
|
|
fi
|
|
|
|
# Activate virtual environment
|
|
source "${VENV_DIR}/bin/activate"
|
|
|
|
# Upgrade pip
|
|
pip install --upgrade pip setuptools wheel
|
|
|
|
# Install PyTorch with CUDA 12.6 support
|
|
echo "Installing PyTorch with CUDA 12.6..."
|
|
pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
|
|
|
|
# Install ComfyUI requirements (upgrade ensures packages match requirements.txt)
|
|
echo "Installing ComfyUI requirements..."
|
|
pip install --upgrade --no-cache-dir -r requirements.txt
|
|
|
|
# Install OpenTelemetry packages if OTEL endpoint is configured
|
|
if [ -n "${OTEL_EXPORTER_OTLP_ENDPOINT}" ]; then
|
|
echo "Installing OpenTelemetry packages..."
|
|
pip install --no-cache-dir opentelemetry-distro opentelemetry-exporter-otlp
|
|
fi
|
|
|
|
# Explicitly ensure frontend package matches requirements.txt version
|
|
echo "Verifying frontend package version..."
|
|
REQUIRED_FRONTEND_VERSION=$(grep "^comfyui-frontend-package==" requirements.txt | cut -d'=' -f3)
|
|
if [ -n "$REQUIRED_FRONTEND_VERSION" ]; then
|
|
echo "Installing frontend package version: $REQUIRED_FRONTEND_VERSION"
|
|
pip install --upgrade --force-reinstall --no-cache-dir "comfyui-frontend-package==${REQUIRED_FRONTEND_VERSION}"
|
|
else
|
|
echo "Warning: Could not determine required frontend version from requirements.txt"
|
|
fi
|
|
|
|
# Optional: Install curated custom nodes (can be enabled via environment variable)
|
|
if [ "${INSTALL_CURATED_NODES:-false}" = "true" ]; then
|
|
echo "Installing curated custom nodes..."
|
|
export COMFYUI_DIR="${COMFYUI_DIR}"
|
|
comfy-node-install \
|
|
https://github.com/city96/ComfyUI-GGUF \
|
|
https://github.com/rgthree/rgthree-comfy \
|
|
https://github.com/ClownsharkBatwing/RES4LYF \
|
|
https://github.com/giriss/comfy-image-saver || echo "Warning: Some custom nodes failed to install"
|
|
fi
|
|
|
|
# Optional: Install Nano Banana node (can be enabled via environment variable)
|
|
if [ "${INSTALL_NANO_BANANA:-false}" = "true" ]; then
|
|
echo "Installing Nano Banana custom node..."
|
|
export COMFYUI_DIR="${COMFYUI_DIR}"
|
|
comfy-node-install https://github.com/ru4ls/ComfyUI_Nano_Banana || echo "Warning: Nano Banana clone failed"
|
|
|
|
# Install Nano Banana dependencies if requirements.txt exists
|
|
if [ -f "${CUSTOM_NODES_DIR}/ComfyUI_Nano_Banana/requirements.txt" ]; then
|
|
echo "Installing Nano Banana dependencies..."
|
|
pip install --no-cache-dir -r "${CUSTOM_NODES_DIR}/ComfyUI_Nano_Banana/requirements.txt" || echo "Warning: Nano Banana dependencies installation failed"
|
|
fi
|
|
fi
|
|
|
|
# Check if OpenTelemetry endpoint is configured
|
|
if [ -n "${OTEL_EXPORTER_OTLP_ENDPOINT}" ]; then
|
|
echo "OpenTelemetry endpoint detected, enabling instrumentation..."
|
|
exec opentelemetry-instrument \
|
|
--traces_exporter otlp \
|
|
--metrics_exporter otlp \
|
|
--logs_exporter otlp \
|
|
python main.py --listen 0.0.0.0 --port 8188 ${COMFYUI_ARGS:-} "$@"
|
|
else
|
|
echo "Starting ComfyUI without OpenTelemetry instrumentation..."
|
|
exec python main.py --listen 0.0.0.0 --port 8188 ${COMFYUI_ARGS:-} "$@"
|
|
fi
|
|
|