Skip to content

GitLab

Unverified Commit 89ede845 authored by Kris Hung's avatar Kris Hung Committed by GitHub
Browse files

fix: Pinned PyTorch version for vLLM container (#2359)

parent bea92eae
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
container/deps/vllm/install_vllm.sh
View file @ 89ede845
...@@ -24,7 +24,7 @@ VLLM_REF="f4135232b9a8c4845f8961fb1cd17581c56ae2ce" ...@@ -24,7 +24,7 @@ VLLM_REF="f4135232b9a8c4845f8961fb1cd17581c56ae2ce"
MAX_JOBS=16 MAX_JOBS=16
INSTALLATION_DIR=/tmp INSTALLATION_DIR=/tmp
ARCH=$(uname -m) ARCH=$(uname -m)
DEEPGEMM_REF="1876566" DEEPGEMM_REF="03d0be3"
FLASHINF_REF="v0.2.8rc1" FLASHINF_REF="v0.2.8rc1"
TORCH_BACKEND="cu128" TORCH_BACKEND="cu128"
...@@ -174,7 +174,7 @@ else ...@@ -174,7 +174,7 @@ else
git clone https://github.com/flashinfer-ai/flashinfer.git --recursive git clone https://github.com/flashinfer-ai/flashinfer.git --recursive
cd flashinfer cd flashinfer
git checkout $FLASHINF_REF git checkout $FLASHINF_REF
python -m pip install -v . uv pip install -v --no-build-isolation .
fi fi
echo "vllm installation completed successfully" echo "vllm installation completed successfully"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment