File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
199200sglang = { git = " https://github.com/sgl-project/sglang.git" , tag = " v0.5.10" , subdirectory = " python" }
200201sglang-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 ]
203205members = [
You can’t perform that action at this time.
0 commit comments