[CPU] toml file update (#17861)
This commit is contained in:
@@ -4,12 +4,6 @@ SHELL ["/bin/bash", "-c"]
|
||||
ARG SGLANG_REPO=https://github.com/sgl-project/sglang.git
|
||||
ARG VER_SGLANG=main
|
||||
|
||||
ARG VER_TORCH=2.9.0
|
||||
ARG VER_TORCHVISION=0.24.0
|
||||
ARG VER_TORCHAUDIO=2.9.0
|
||||
ARG VER_TORCHAO=0.14.1
|
||||
ARG VER_TRITON=3.5.0
|
||||
|
||||
RUN apt-get update && \
|
||||
apt-get full-upgrade -y && \
|
||||
DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
|
||||
@@ -46,8 +40,6 @@ RUN source $HOME/.local/bin/env && \
|
||||
cd python && \
|
||||
cp pyproject_cpu.toml pyproject.toml && \
|
||||
uv pip install . && \
|
||||
uv pip install torch==${VER_TORCH} torchvision==${VER_TORCHVISION} torchaudio==${VER_TORCHAUDIO} torchao==${VER_TORCHAO} triton==${VER_TRITON} --force-reinstall && \
|
||||
uv pip install tabulate && \
|
||||
cd ../sgl-kernel && \
|
||||
cp pyproject_cpu.toml pyproject.toml && \
|
||||
uv pip install .
|
||||
|
||||
Reference in New Issue
Block a user