diff --git a/pyproject.toml b/pyproject.toml index b102d93c8f..3c494ed1dd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -112,7 +112,7 @@ override-dependencies = [ "torch; sys_platform == 'never'", "torchvision; sys_platform == 'never'", "triton; sys_platform == 'never'", - "transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@5671fd3675906cda1ade26c24a65d3dedd88eb89", + "transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@287770466f0f4433052260a765db5ff7b8be1320", "mlflow>=3.5.0", "cryptography>=43.0.0,<47", "nvidia-modelopt>=0.42.0a0", diff --git a/uv.lock b/uv.lock index c2c276790a..71c67be6c4 100644 --- a/uv.lock +++ b/uv.lock @@ -34,7 +34,7 @@ overrides = [ { name = "nvidia-modelopt", specifier = ">=0.42.0a0", index = "https://pypi.nvidia.com/" }, { name = "torch", marker = "sys_platform == 'never'" }, { name = "torchvision", marker = "sys_platform == 'never'" }, - { name = "transformer-engine", git = "https://github.com/NVIDIA/TransformerEngine.git?rev=5671fd3675906cda1ade26c24a65d3dedd88eb89" }, + { name = "transformer-engine", git = "https://github.com/NVIDIA/TransformerEngine.git?rev=287770466f0f4433052260a765db5ff7b8be1320" }, { name = "triton", marker = "sys_platform == 'never'" }, { name = "urllib3", specifier = ">=2.6.3" }, ] @@ -7813,8 +7813,8 @@ wheels = [ [[package]] name = "transformer-engine" -version = "2.12.0+5671fd36" -source = { git = "https://github.com/NVIDIA/TransformerEngine.git?rev=5671fd3675906cda1ade26c24a65d3dedd88eb89#5671fd3675906cda1ade26c24a65d3dedd88eb89" } +version = "2.13.0+28777046" +source = { git = "https://github.com/NVIDIA/TransformerEngine.git?rev=287770466f0f4433052260a765db5ff7b8be1320#287770466f0f4433052260a765db5ff7b8be1320" } dependencies = [ { name = "einops" }, { name = "importlib-metadata" },