Skip to content

Commit 8704af9

Browse files
Update dependencies
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
1 parent 7143ebf commit 8704af9

2 files changed

Lines changed: 75 additions & 70 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ vllm = [
110110
{ index = "pytorch-cu130", marker = "python_version < '3.9' and platform_machine == 'x86_64'" },
111111
{ index = "pypi", marker = "platform_machine == 'aarch64'" },
112112
]
113-
megatron-bridge = { git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge.git", rev = "859a18d001eca4423ad79e7771632d839f50fa77" }
113+
megatron-bridge = { git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge.git", rev = "97e8dba4060ae74a110b9da6255eb0a66b016b2d" }
114114
nvidia-resiliency-ext = { git = "https://github.com/NVIDIA/nvidia-resiliency-ext.git", rev = "b2bb3d728a18795807d9f76c535e005a609a1b01" }
115115
# nemo-toolkit = { git = "https://github.com/NVIDIA/NeMo.git", rev = "main" }
116116

@@ -124,6 +124,7 @@ no-build-isolation-package = [
124124
"causal-conv1d",
125125
"nv-grouped-gemm",
126126
]
127+
127128
# Always apply the build group since dependencies like TE/mcore/nemo-run require build dependencies
128129
# and this lets us assume they are implicitly installed with a simply `uv sync`. Ideally, we'd
129130
# avoid including these in the default dependency set, but for now it's required.
@@ -134,6 +135,7 @@ default-groups = ["linting", "build", "test"]
134135
# --link-mode=symlink (fastest option when uv cache and venv on different file-system; caveat: venv is brittle since it depends on the environment/container)
135136
link-mode = "copy"
136137
conflicts = [[{ extra = "vllm" }, { extra = "trt-onnx" }]]
138+
extra-build-dependencies = { fast-hadamard-transform = ["setuptools", "torch"] }
137139
override-dependencies = [
138140
"cachetools>=5.0.0",
139141
"torch; sys_platform == 'never'",

0 commit comments

Comments
 (0)