feat: add virtual environment support for ComfyUI
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>
This commit is contained in:
23
arty.yml
23
arty.yml
@@ -229,19 +229,34 @@ scripts:
|
|||||||
cd $COMFYUI_ROOT && git pull
|
cd $COMFYUI_ROOT && git pull
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Create virtual environment for ComfyUI
|
||||||
|
echo "Creating virtual environment for ComfyUI..."
|
||||||
|
cd $COMFYUI_ROOT
|
||||||
|
if [ ! -d "venv" ]; then
|
||||||
|
python3 -m venv venv
|
||||||
|
fi
|
||||||
|
source venv/bin/activate
|
||||||
|
|
||||||
# Install ComfyUI dependencies
|
# Install ComfyUI dependencies
|
||||||
echo "Installing ComfyUI dependencies..."
|
echo "Installing ComfyUI dependencies..."
|
||||||
sudo pip3 install -r $COMFYUI_ROOT/requirements.txt
|
pip install -r requirements.txt
|
||||||
|
|
||||||
# Install additional ComfyUI dependencies
|
# Install additional ComfyUI dependencies
|
||||||
if [ -f "comfyui/requirements.txt" ]; then
|
if [ -f "$AI_ROOT/comfyui/requirements.txt" ]; then
|
||||||
echo "Installing additional ComfyUI dependencies..."
|
echo "Installing additional ComfyUI dependencies..."
|
||||||
sudo pip3 install -r comfyui/requirements.txt
|
pip install -r $AI_ROOT/comfyui/requirements.txt
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Install common extension dependencies
|
||||||
|
echo "Installing common extension dependencies..."
|
||||||
|
pip install GitPython opencv-python-headless diffusers insightface onnxruntime
|
||||||
|
|
||||||
|
deactivate
|
||||||
|
cd $AI_ROOT
|
||||||
|
|
||||||
# Create model directory structure
|
# Create model directory structure
|
||||||
echo "Creating ComfyUI model directories..."
|
echo "Creating ComfyUI model directories..."
|
||||||
mkdir -p $COMFYUI_ROOT/models/{checkpoints,unet,vae,loras,clip,clip_vision,controlnet,ipadapter,embeddings,upscale_models,video_models,animatediff_models,animatediff_motion_lora,audio_models,configs,diffusers,diffusion_models}
|
mkdir -p $COMFYUI_ROOT/models/{checkpoints,unet,vae,loras,clip,clip_vision,controlnet,ipadapter,embeddings,upscale_models,video_models,animatediff_models,animatediff_motion_lora,audio_models,configs,diffusers,diffusion_models,musicgen}
|
||||||
|
|
||||||
# Create symlink to huggingface cache
|
# Create symlink to huggingface cache
|
||||||
echo "Creating symlink to HuggingFace cache..."
|
echo "Creating symlink to HuggingFace cache..."
|
||||||
|
|||||||
@@ -16,12 +16,20 @@ export TQDM_DISABLE=1
|
|||||||
# Navigate to ComfyUI directory
|
# Navigate to ComfyUI directory
|
||||||
cd "${COMFYUI_DIR}" || exit 1
|
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 "Starting ComfyUI on port 8188..."
|
||||||
echo "Access at: http://localhost:8188"
|
echo "Access at: http://localhost:8188"
|
||||||
echo "Using HuggingFace cache: ${HF_CACHE}"
|
echo "Using HuggingFace cache: ${HF_CACHE}"
|
||||||
|
|
||||||
# Start ComfyUI with GPU support
|
# Start ComfyUI with GPU support
|
||||||
python3 main.py \
|
python main.py \
|
||||||
--listen 0.0.0.0 \
|
--listen 0.0.0.0 \
|
||||||
--port 8188 \
|
--port 8188 \
|
||||||
--enable-cors-header \
|
--enable-cors-header \
|
||||||
|
|||||||
Reference in New Issue
Block a user