Skip to content

Commit 8a955c0

Browse files
committed
chore: Bump TE to latest 2.14
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent 3b9c627 commit 8a955c0

2 files changed

Lines changed: 28 additions & 28 deletions

File tree

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ no-build-isolation-package = [
189189
link-mode = "copy"
190190
conflicts = [[{ extra = "lts" }, { extra = "dev" }]]
191191
# We don't define override-dependencies globally but rather locally where we need it.
192-
# For instance, when installing into a PyTorch base image, we don't want to install torch,
192+
# For instance, when installing into a PyTorch base image, we don't want to install torch,
193193
# torchvision, and triton.
194194
override-dependencies = [
195195
"torch; sys_platform == 'never'",
@@ -202,7 +202,7 @@ override-dependencies = [
202202
flash_mla = [
203203
{ git = "https://github.com/deepseek-ai/FlashMLA", rev = "9edee0c022cd0938148a18e334203b0aab43aa19" },
204204
]
205-
transformer-engine = { git = "https://github.com/NVIDIA/TransformerEngine.git", rev = "71bbefbf153418f943640df0f7373625dc93fa46" }
205+
transformer-engine = { git = "https://github.com/NVIDIA/TransformerEngine.git", rev = "366798ef8a0a00d8f2c1650d11e7e623d7c33e26" }
206206
nemo-run = { git = "https://github.com/NVIDIA-NeMo/Run.git", rev = "01a9a8ba360f7b2908728ad0516e0ad9d936966d" }
207207
emerging_optimizers = { git = "https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git", rev = "v0.1.0" }
208208
nvidia-resiliency-ext = { git = "https://github.com/NVIDIA/nvidia-resiliency-ext.git", rev = "15a851565a4ce846c04431ecb0cf09903ab4837e" }

uv.lock

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

0 commit comments

Comments
 (0)