Skip to content

Commit b6c4a38

Browse files
committed
Merge branch 'main' of https://github.com/openpipe/art
2 parents 14e3d19 + 2827c4b commit b6c4a38

2 files changed

Lines changed: 640 additions & 167 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ backend = [
3838
"pytest>=8.4.1",
3939
"nbmake>=1.5.5",
4040
"gql<4",
41-
"vllm==0.15.1 ; sys_platform == 'linux'",
41+
"vllm==0.17.0 ; sys_platform == 'linux'",
4242
]
4343
megatron = [
4444
"torch>=2.8.0",
@@ -120,8 +120,8 @@ asyncio_mode = "auto"
120120
[tool.uv]
121121
required-version = ">=0.6.15"
122122
# Override numpy to <2.0 for compatibility with megatron-core in the training
123-
# environment. vLLM 0.15.1 pulls opencv-python-headless>=4.13 which wants
124-
# numpy>=2 on Python 3.9+, but megatron-core requires numpy<2.
123+
# environment. vLLM pulls opencv-python-headless>=4.13 which wants numpy>=2 on
124+
# Python 3.9+, but megatron-core requires numpy<2.
125125
override-dependencies = ["transformer-engine>=2.11.0", "numpy<2"]
126126
# Keep apex build isolation enabled so uv can inject torch from
127127
# `extra-build-dependencies` during lock/sync on non-GPU client machines.

0 commit comments

Comments
 (0)