#!/bin/sh

set -e

pip install --upgrade pip
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
pip install -r requirements.txt
MAX_JOBS=4 pip install --no-build-isolation flash-attn==2.1.1