File size: 1,091 Bytes
ec0a9aa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | #!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
ENV_DIR="${1:-$ROOT_DIR/DreamDojo-radial/.venv}"
PYTHON_BIN="${PYTHON_BIN:-python3.12}"
if ! command -v "$PYTHON_BIN" >/dev/null 2>&1; then
echo "Missing $PYTHON_BIN. Set PYTHON_BIN=/path/to/python3.12 and rerun." >&2
exit 1
fi
"$PYTHON_BIN" -m venv "$ENV_DIR"
source "$ENV_DIR/bin/activate"
python -m pip install --upgrade pip setuptools wheel
python -m pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
python -m pip install -r "$ROOT_DIR/video_gen_physics/models/dreamgen/requirements.txt"
python -m pip install -r "$ROOT_DIR/video_gen_physics/methods/sparse_attention/radial-attention/requirements.txt"
python -m pip install flash-attn --no-build-isolation
python -m pip install flashinfer-python -i https://flashinfer.ai/whl/cu124/torch2.5/
cat <<EOF
Radial environment created at: $ENV_DIR
Activate it with:
source "$ENV_DIR/bin/activate"
Then run DreamGen with:
export PYTHONPATH="./models/dreamgen"
EOF
|