Skip to content

Commit 56cbd1b

Browse files
committed
Update MCore
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent fa6fd63 commit 56cbd1b

2 files changed

Lines changed: 257 additions & 49 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,8 @@ vllm = [
114114
{ index = "pytorch-cu130", marker = "python_version < '3.9' and platform_machine == 'x86_64'" },
115115
{ index = "pypi", marker = "platform_machine == 'aarch64'" },
116116
]
117-
megatron-bridge = { git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge.git", rev = "0465189601951c14ad592bd8eefbb673929e25bc" }
117+
megatron-bridge = { git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge.git", rev = "4e3dbd2952ea1feca388f679ed14675527d2eef8" }
118+
megatron-core = { git = "https://github.com/NVIDIA/Megatron-LM.git", rev = "b213e10a5fb1f3700ec227db71dd7a8a75dffce7" }
118119
# transformer-engine = { git = "https://github.com/NVIDIA/TransformerEngine.git", rev = "c188b533cc3721ca9c6bbfd26148f5cf60108c25" } # on release_v2.11.0
119120
# nemo-toolkit = { git = "https://github.com/NVIDIA/NeMo.git", rev = "main" }
120121

@@ -149,7 +150,7 @@ override-dependencies = [
149150
"transformer-engine-cu13>=2.10.0a0,<2.13.0; sys_platform != 'darwin'",
150151
"transformer-engine-cu12; sys_platform == 'never'",
151152
"open-clip-torch>=3.2.0",
152-
"transformers==4.57.5"
153+
"mamba-ssm>=2.3.0,<2.4.0"
153154
]
154155
prerelease = "allow"
155156

0 commit comments

Comments
 (0)