Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env bash
- set -e
- # Install nvm and Node.js
- export NVM_DIR="$HOME/.nvm"
- curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.5/install.sh | bash
- source "$NVM_DIR/nvm.sh"
- nvm install 18
- nvm use 18
- # Install uv and activate
- curl -LsSf https://astral.sh/uv/install.sh | sh
- source "$HOME/.local/bin/env"
- # Clone repo
- git clone https://github.com/ostris/ai-toolkit.git "$HOME/ai-toolkit"
- cd "$HOME/ai-toolkit"
- git submodule update --init --recursive
- # Create and activate venv
- uv venv .venv
- source .venv/bin/activate
- # Install Python deps
- uv pip install --upgrade pip
- uv pip install --no-cache-dir torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128
- uv pip install --no-cache-dir -r requirements.txt --torch-backend=auto
- uv pip install --upgrade accelerate transformers diffusers huggingface_hub
- # Install and run UI
- cd ui
- npm install
- npm run build_and_start
Advertisement
Add Comment
Please, Sign In to add comment