Skip to content

Commit 2d19863

Browse files
committed
fix with nixl
Signed-off-by: jthomson04 <jwillthomson19@gmail.com>
1 parent 93e484e commit 2d19863

2 files changed

Lines changed: 3786 additions & 2475 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ dependencies = [
5656
"nvidia-cudnn-cu12==9.19.0.56", # for transformer-engine no build isolation
5757
"ai-dynamo @ file:///opt/nemo-rl/ai_dynamo-1.0.0-py3-none-any.whl",
5858
"ai-dynamo-runtime @ file:///opt/nemo-rl/ai_dynamo_runtime-1.0.0-cp310-abi3-manylinux_2_39_x86_64.whl",
59+
"nixl==0.10.1",
5960
"aiconfigurator[webapp]",
6061
"aiofiles<=25.1.0",
6162
"filterpy==1.4.5",
@@ -65,7 +66,7 @@ dependencies = [
6566
"prometheus-api-client==0.6.0",
6667
"prophet==1.2.1",
6768
"scikit-learn==1.7.2",
68-
"scipy<1.14.0",
69+
"scipy>=1.14.0,<2.0",
6970
# nvidia-resiliency-ext removed: no Python 3.13 wheels available (v0.5.0 only has cp310-cp312)
7071
]
7172

@@ -198,6 +199,7 @@ emerging-optimizers = { git = "https://github.com/NVIDIA-NeMo/Emerging-Optimizer
198199
# Official sglang v0.5.10 with transformers v5 support
199200
sglang = { git = "https://github.com/sgl-project/sglang.git", tag = "v0.5.10", subdirectory = "python" }
200201
sglang-kernel = { git = "https://github.com/sgl-project/sglang.git", tag = "v0.5.10", subdirectory = "sgl-kernel" }
202+
aiconfigurator = { git = "https://github.com/jthomson04/aiconfigurator.git", rev = "4ed8604c43f221b60aa501cb218722fd7029fdb6" }
201203

202204
[tool.uv.workspace]
203205
members = [

0 commit comments

Comments
 (0)