Sebastian Krüger
4dd608a67d
Changes: - Create venv for ComfyUI in setup/comfyui-base script - Install extension dependencies: GitPython, opencv-python-headless, diffusers, insightface, onnxruntime - Update start.sh to activate venv before running - Add musicgen model directory This fixes import errors for custom nodes: - ComfyUI-Manager (needs GitPython) - ComfyUI-Impact-Pack (needs opencv) - ComfyUI-VideoHelperSuite (needs opencv) - ComfyUI-CogVideoXWrapper (needs diffusers) - ComfyUI-Inspire-Pack (needs insightface, onnxruntime) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
39 lines
889 B
Bash
39 lines
889 B
Bash
#!/bin/bash
|
|
#
|
|
# ComfyUI Startup Script
|
|
# Starts ComfyUI server on port 8188
|
|
#
|
|
|
|
WORKSPACE_DIR="${WORKSPACE_DIR:-/workspace}"
|
|
COMFYUI_DIR="${WORKSPACE_DIR}/ComfyUI"
|
|
HF_CACHE="${WORKSPACE_DIR}/huggingface_cache"
|
|
|
|
# Set environment variables
|
|
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
|
export HF_HOME="${HF_CACHE}"
|
|
export TQDM_DISABLE=1
|
|
|
|
# Navigate to ComfyUI directory
|
|
cd "${COMFYUI_DIR}" || exit 1
|
|
|
|
# Activate virtual environment
|
|
if [ -d "venv" ]; then
|
|
echo "Activating ComfyUI virtual environment..."
|
|
source venv/bin/activate
|
|
else
|
|
echo "WARNING: venv not found, using system Python"
|
|
fi
|
|
|
|
echo "Starting ComfyUI on port 8188..."
|
|
echo "Access at: http://localhost:8188"
|
|
echo "Using HuggingFace cache: ${HF_CACHE}"
|
|
|
|
# Start ComfyUI with GPU support
|
|
python main.py \
|
|
--listen 0.0.0.0 \
|
|
--port 8188 \
|
|
--enable-cors-header \
|
|
--preview-method auto
|
|
|
|
echo "ComfyUI stopped"
|