Skip to content

Commit b778253

Browse files
committed
address comments
1 parent 987e26f commit b778253

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

container/deps/vllm/install_vllm.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,17 +147,16 @@ if [ "$ARCH" = "arm64" ]; then
147147
fi
148148
else
149149
echo "Installing vllm for AMD64 architecture"
150-
uv pip install torch==2.7.1+cu128 torchaudio==2.7.1 torchvision==0.22.1 --index-url https://download.pytorch.org/whl/cu128
151150
VLLM_TEMP_DIR=/tmp/vllm/wheel/$VLLM_REF
152151
mkdir -p $VLLM_TEMP_DIR
153152
REMOTE_WHEEL_URL=https://vllm-wheels.s3.us-west-2.amazonaws.com/${VLLM_REF}/vllm-1.0.0.dev-cp38-abi3-manylinux1_x86_64.whl
154153
export VLLM_PRECOMPILED_WHEEL_LOCATION=$VLLM_TEMP_DIR/vllm-1.0.0.dev-cp38-abi3-manylinux1_x86_64.whl
155154
rm -rf $VLLM_PRECOMPILED_WHEEL_LOCATION || true
156155
curl -fS --retry 3 -L "$REMOTE_WHEEL_URL" -o "$VLLM_PRECOMPILED_WHEEL_LOCATION"
157156
if [ "$EDITABLE" = "true" ]; then
158-
VLLM_USE_PRECOMPILED=1 uv pip install -e . --torch-backend=$TORCH_BACKEND
157+
uv pip install -e . --torch-backend=$TORCH_BACKEND
159158
else
160-
VLLM_USE_PRECOMPILED=1 uv pip install . --torch-backend=$TORCH_BACKEND
159+
uv pip install . --torch-backend=$TORCH_BACKEND
161160
fi
162161
rm -rf $VLLM_PRECOMPILED_WHEEL_LOCATION
163162
fi

0 commit comments

Comments
 (0)