Skip to content

Commit 5c475f5

Browse files
authored
[chore] move megatron-core from dev latest to main latest, bump megatron-bridge to latest (NovaSky-AI#1762)
This PR makes the following bumps megatron_core from `cefc2520158c7ceba3f9adbe4b547a6f7a118da1` (latest dev branch as of 6/8/26) to `71e418ea7d7b3a6c9a53238c543c3e0b43e11026` (latest main branch as of 6/8/26. megatron-bridge from `8382dc343b07b068a827ca20bae860633df3baa0` to `91a15142a4b4442a8d46ab539d1b923bd08570d0` (latest main 6/8) Megatron-Bridge has upstreamed code to the main branch that isn't on the dev branch that is needed to use Megatron-Bridge (NVIDIA-NeMo/Megatron-Bridge#3988) Since sequence packing with GDN is now supported on main, we can move back over to the latest commit on the main branch: NVIDIA/Megatron-LM#2645
1 parent a89f88f commit 5c475f5

2 files changed

Lines changed: 18 additions & 20 deletions

File tree

pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ override-dependencies = [
222222
"nvidia-resiliency-ext; sys_platform == 'never'",
223223
"transformer-engine[pytorch]==2.11.0; sys_platform == 'linux'",
224224
"transformers>=5.6.1,<=5.8.0; sys_platform == 'linux'",
225-
"megatron-core>=0.16.0; sys_platform == 'linux'",
225+
"megatron-core>=0.16.0; sys_platform == 'linux' and python_version >= '3.12'",
226226
"ml_dtypes>=0.5.0; sys_platform == 'linux'",
227227
]
228228

@@ -296,9 +296,9 @@ torchvision = [
296296
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
297297
]
298298
harbor = { git = "https://github.com/laude-institute/harbor", rev = "a85628c803dc1713fcd26d2bb1908cde69d62317" }
299-
megatron-bridge = {git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge", rev = "8382dc343b07b068a827ca20bae860633df3baa0", marker = "sys_platform == 'linux'"}
300-
# megatron-core dev branch: https://github.com/NVIDIA/Megatron-LM/tree/dev latest as of 6/5/26
301-
megatron-core = {git = "https://github.com/NVIDIA/Megatron-LM", rev = "cefc2520158c7ceba3f9adbe4b547a6f7a118da1", marker = "sys_platform == 'linux'"}
299+
megatron-bridge = {git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge", rev = "91a15142a4b4442a8d46ab539d1b923bd08570d0", marker = "sys_platform == 'linux'"}
300+
# megatron-core main branch: https://github.com/NVIDIA/Megatron-LM/tree/main latest as of 6/8/26
301+
megatron-core = {git = "https://github.com/NVIDIA/Megatron-LM", rev = "71e418ea7d7b3a6c9a53238c543c3e0b43e11026", marker = "sys_platform == 'linux'"}
302302
# NOTE (sumanthrh): This custom wheel of vllm-router includes a fix for /chat/completions endpoint: https://github.com/vllm-project/router/pull/162 on top of 0.1.14
303303
# TODO (sumanthrh): Remove this after 0.1.15 vllm-router release
304304
vllm-router = { url = "https://github.com/SumanthRH/router/releases/download/0.1.14.post1/vllm_router-0.1.14-cp38-abi3-manylinux_2_35_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }

0 commit comments

Comments
 (0)