30 lines
1.2 KiB
Bash
30 lines
1.2 KiB
Bash
#!/bin/bash
|
|
|
|
# install_deps.sh - Installs dependencies for ComfyUI_LocalLLMNodes
|
|
|
|
# Exit immediately if a command exits with a non-zero status.
|
|
set -e
|
|
|
|
echo "===== Installing ComfyUI_LocalLLMNodes Dependencies ====="
|
|
|
|
# --- Install core dependencies from requirements.txt ---
|
|
# This installs transformers, torch, bitsandbytes (if listed), and llama-cpp-python (CPU version potentially)
|
|
echo "1/3: Installing core dependencies from requirements.txt..."
|
|
pip install -r requirements.txt
|
|
|
|
# --- Install llama-cpp-python with CUDA support ---
|
|
# This overwrites/ensures the GPU-accelerated version is installed
|
|
echo "2/3: Installing llama-cpp-python with CUDA support..."
|
|
echo " (This might take a while and download/build components...)"
|
|
# Set environment variable and install llama-cpp-python
|
|
# Adjust cu118/cu121/cu124 based on your CUDA version (check `nvcc --version`)
|
|
CMAKE_ARGS="-DGGML_CUDA=on" pip install llama-cpp-python --force-reinstall --no-cache-dir
|
|
|
|
# --- Completion ---
|
|
echo "3/3: Installation process completed."
|
|
echo "===== Installation Finished ====="
|
|
echo ""
|
|
echo "Please ensure you have the NVIDIA CUDA toolkit installed"
|
|
echo "and that your environment is set up correctly for ComfyUI."
|
|
echo "You can now start ComfyUI."
|