Skip to content

Commit 14ff6f0

Browse files
Eta0Eta0
authored
Merge pull request #117 from coreweave/es/vllm-torch-2.9.1
build(vllm-tensorizer): Update base images to ones with PyTorch v2.9.1
2 parents 19331b7 + fd84cfa commit 14ff6f0

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

‎.github/configurations/vllm-tensorizer.yml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,6 @@ vllm-commit:
33
flashinfer-commit:
44
- 'v0.5.2'
55
builder-base-image:
6-
- 'ghcr.io/coreweave/ml-containers/torch:8a60b2d-nccl-cuda12.9.1-ubuntu22.04-nccl2.28.3-1-torch2.8.0-vision0.23.0-audio2.8.0-abi1'
6+
- 'ghcr.io/coreweave/ml-containers/torch:19331b7-nccl-cuda12.9.1-ubuntu22.04-nccl2.29.2-1-torch2.9.1-vision0.24.1-audio2.9.1-abi1'
77
final-base-image:
8-
- 'ghcr.io/coreweave/ml-containers/torch:8a60b2d-nccl-cuda12.9.1-ubuntu22.04-nccl2.28.3-1-torch2.8.0-vision0.23.0-audio2.8.0-abi1'
8+
- 'ghcr.io/coreweave/ml-containers/torch:19331b7-nccl-cuda12.9.1-ubuntu22.04-nccl2.29.2-1-torch2.9.1-vision0.24.1-audio2.9.1-abi1'

‎vllm-tensorizer/Dockerfile‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
ARG BUILDER_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch:8a60b2d-nccl-cuda12.9.1-ubuntu22.04-nccl2.28.3-1-torch2.8.0-vision0.23.0-audio2.8.0-abi1"
2-
ARG FINAL_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch:8a60b2d-nccl-cuda12.9.1-ubuntu22.04-nccl2.28.3-1-torch2.8.0-vision0.23.0-audio2.8.0-abi1"
1+
ARG BUILDER_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch:19331b7-nccl-cuda12.9.1-ubuntu22.04-nccl2.29.2-1-torch2.9.1-vision0.24.1-audio2.9.1-abi1"
2+
ARG FINAL_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch:19331b7-nccl-cuda12.9.1-ubuntu22.04-nccl2.29.2-1-torch2.9.1-vision0.24.1-audio2.9.1-abi1"
33

44
FROM scratch AS freezer
55
WORKDIR /

0 commit comments

Comments
 (0)