Skip to content

Commit 6b63ed9

Browse files
authored
Replace megatron-hub with megatron-bridge (#27)
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent 626cc15 commit 6b63ed9

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -96,12 +96,12 @@ test = ["pytest", "pytest-mock", "coverage", "click", "pytest-httpserver"]
9696
nemo-legacy = [
9797
"nemo-toolkit[automodel,common-only,nlp-only,eval,multimodal-only]",
9898
]
99-
megatron-hub-pre = ["megatron-hub"]
99+
megatron-bridge-pre = ["megatron-bridge"]
100100
nemo-export-deploy-pre = ["nemo-export-deploy"]
101101

102102
[tool.uv.sources]
103103
# Remove this
104-
megatron-hub = { git = "https://github.com/NVIDIA-NeMo/Megatron-Hub.git", rev = "cc2a60c3be11549f070fe1d7b6324a3dae51c6e0" }
104+
megatron-bridge = { git = "https://github.com/NVIDIA-NeMo/Megatron-Bridge.git", rev = "958fb9b7da4eb1506142a122965fa5424cfdc8a2" }
105105
nemo-toolkit = { git = "https://github.com/NVIDIA/NeMo.git", rev = "5e63ca3c1631bb156187705f9c4cf643a67f9c05" }
106106
megatron-core = { git = "https://github.com/NVIDIA/Megatron-LM.git", rev = "7b296e0553ec19723d77a2310cb9808ab373eb2a" }
107107
nemo-export-deploy = { git = "https://github.com/NVIDIA-NeMo/Export-Deploy.git", rev = "b2c0f1912cb69ed9e843e5a6299c1ea53b6d5aee" }
@@ -120,7 +120,7 @@ default-groups = ["linting", "build", "test"]
120120
link-mode = "copy"
121121
override-dependencies = [
122122
"urllib3>1.27.0",
123-
"tiktoken>=0.9.0", # because nemo-toolkit and megatron-hub disagree on tiktoken, we need to pin it here,
123+
"tiktoken>=0.9.0", # because nemo-toolkit and megatron-bridge disagree on tiktoken, we need to pin it here,
124124
]
125125
prerelease = "allow"
126126

uv.lock

Lines changed: 5 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)