From 0e8db24417c9a88990120c9c0545ddce60cc1bb5 Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Mon, 2 Dec 2024 00:30:14 +0000 Subject: [PATCH] Update transformer-engine[pytorch] requirement Updates the requirements on [transformer-engine[pytorch]](https://github.com/NVIDIA/TransformerEngine) to permit the latest version. - [Release notes](https://github.com/NVIDIA/TransformerEngine/releases) - [Commits](https://github.com/NVIDIA/TransformerEngine/compare/v1.11...v1.12) --- updated-dependencies: - dependency-name: transformer-engine[pytorch] dependency-type: direct:development ... Signed-off-by: dependabot[bot] --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 6aa1553b5d..d16342a9b1 100644 --- a/setup.py +++ b/setup.py @@ -124,7 +124,7 @@ ] extra_deps['te'] = [ - 'transformer-engine[pytorch]>=1.11.0,<1.12', + 'transformer-engine[pytorch]>=1.11.0,<1.13', ] extra_deps['databricks-serverless'] = {