chore: bump v0.4.7 (#7038)

This commit is contained in:
Yineng Zhang
2025-06-10 01:56:20 -07:00
committed by GitHub
parent fcde67b016
commit 4f723edd3b
8 changed files with 19 additions and 15 deletions

View File

@@ -37,9 +37,13 @@ RUN python3 -m pip install --upgrade pip setuptools wheel html5lib six \
else \
echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1; \
fi \
&& python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu${CUINDEX} \
&& if [ "$CUDA_VERSION" = "12.4.1" ]; then \
python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu126; \
else \
python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu${CUINDEX}; \
fi \
&& cd sglang \
&& python3 -m pip --no-cache-dir install -e "python[${BUILD_TYPE}]" --find-links https://flashinfer.ai/whl/cu${CUINDEX}/torch2.6/flashinfer-python \
&& python3 -m pip --no-cache-dir install -e "python[${BUILD_TYPE}]" \
&& if [ "$CUDA_VERSION" = "12.8.1" ]; then \
python3 -m pip install nvidia-nccl-cu12==2.26.2.post1 --force-reinstall --no-deps; \
fi