Skip to content

Commit

Permalink
upd
Browse files Browse the repository at this point in the history
  • Loading branch information
zhyncs committed Feb 1, 2025
1 parent b1cf3fa commit 9ba0ee7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/ci_install_dependency.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,13 @@ set -euxo pipefail
# Install the dependency in CI.

# Use repo from environment variable, passed from GitHub Actions
FLASHINFER_REPO="${FLASHINFER_REPO:-https://flashinfer.ai/whl/cu124/torch2.4/flashinfer}"
FLASHINFER_REPO="${FLASHINFER_REPO:-https://flashinfer.ai/whl/cu124/torch2.5/flashinfer}"

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
bash "${SCRIPT_DIR}/killall_sglang.sh"

pip install --upgrade pip
pip install -e "python[all]" --find-links https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/
pip install -e "python[all]" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer/

# Force reinstall flashinfer and torch_memory_saver
pip install flashinfer==0.2.0.post2 --find-links ${FLASHINFER_REPO} --force-reinstall --no-deps
Expand Down

0 comments on commit 9ba0ee7

Please sign in to comment.