From d0593973d386e47418725f209e9abaa6d8b7ed3a Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Mon, 9 Dec 2024 00:29: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/commits) --- 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 57acaddf5a..3d48039e1b 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.14', ] extra_deps['databricks-serverless'] = {