File tree Expand file tree Collapse file tree 2 files changed +5
-6
lines changed
Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Original file line number Diff line number Diff line change 11torch==2.8.0
22torchvision
33peft==0.15.1
4- protobuf==3.20.3
4+ protobuf==4.25.1
55transformers==4.55.2
66hf-transfer
77zstandard
@@ -20,7 +20,7 @@ retrying
2020opencv-contrib-python-headless
2121safetensors
2222scipy==1.16.0
23- onnx==1.16 .0
23+ onnx==1.19 .0
2424sentence_transformers
2525onnxruntime-gpu==1.20.0
2626autoawq
3131ninja
3232peft
3333llmcompressor
34- https://publish.djl.ai/sm- vllm/vllm- 0.10.2-cp38-abi3-linux_x86_64.whl
35- xgrammar==0.1.23
34+ vllm== 0.10.2
35+ xgrammar
3636flashinfer-python==0.2.5
Original file line number Diff line number Diff line change @@ -89,8 +89,7 @@ RUN scripts/patch_oss_dlc.sh python \
8989 && apt-get clean -y && rm -rf /var/lib/apt/lists/*
9090
9191COPY lmi-container-requirements.txt ./requirements.txt
92- RUN pip3 install torch==2.6.0 torchvision \
93- && pip3 install flashinfer-python -i https://flashinfer.ai/whl/cu126/torch2.6/ \
92+ RUN pip3 install torch==2.8.0 torchvision \
9493 && pip3 install -r requirements.txt \
9594 && pip3 install ${djl_converter_wheel} --no-deps
9695
You can’t perform that action at this time.
0 commit comments