diff --git a/pyproject.toml b/pyproject.toml index 7b0aa1f684..9ba645f096 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -119,7 +119,7 @@ override-dependencies = [ "torch; sys_platform == 'never'", "torchvision; sys_platform == 'never'", "triton; sys_platform == 'never'", - "transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@f031cf87bd054c7558b887df7bed93975456667f", + "transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@release_v2.15", "mlflow>=3.9.0", # To address CVE-2025-15031 "cachetools>=5.0.0", "cryptography>=43.0.0,<47", diff --git a/uv.lock b/uv.lock index 9aa1b95135..33c0217c60 100644 --- a/uv.lock +++ b/uv.lock @@ -20,7 +20,7 @@ overrides = [ { name = "onnx", marker = "python_full_version < '3.13'", specifier = ">=1.21.0" }, { name = "torch", marker = "sys_platform == 'never'" }, { name = "torchvision", marker = "sys_platform == 'never'" }, - { name = "transformer-engine", git = "https://github.com/NVIDIA/TransformerEngine.git?rev=f031cf87bd054c7558b887df7bed93975456667f" }, + { name = "transformer-engine", git = "https://github.com/NVIDIA/TransformerEngine.git?rev=release_v2.15" }, { name = "triton", marker = "sys_platform == 'never'" }, { name = "urllib3", specifier = ">=2.6.3" }, ] @@ -5034,8 +5034,8 @@ wheels = [ [[package]] name = "transformer-engine" -version = "2.14.0+f031cf87" -source = { git = "https://github.com/NVIDIA/TransformerEngine.git?rev=f031cf87bd054c7558b887df7bed93975456667f#f031cf87bd054c7558b887df7bed93975456667f" } +version = "2.15.0+3378ef15" +source = { git = "https://github.com/NVIDIA/TransformerEngine.git?rev=release_v2.15#3378ef15d6581d20b2fda62918827f279ccb55f3" } dependencies = [ { name = "einops" }, { name = "importlib-metadata" },