From 3542e0b8440383f017fd16670511a579ef6e5aa9 Mon Sep 17 00:00:00 2001
From: Kyle Mylonakis
 <122286752+KyleMylonakisProtopia@users.noreply.github.com>
Date: Thu, 12 Jun 2025 15:34:13 +0100
Subject: [PATCH] build: :pushpin: Remove upper bound on PyTorch (#38789)

build: :pushpin: remove upper bound on torch dependency as issue which originally resulted in the pin has been released in torch 2.7.1
---
 setup.py                                      | 2 +-
 src/transformers/dependency_versions_table.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 04942c88b4..d47ccb197c 100644
--- a/setup.py
+++ b/setup.py
@@ -189,7 +189,7 @@ _deps = [
     "tiktoken",
     "timm<=1.0.11",
     "tokenizers>=0.21,<0.22",
-    "torch>=2.1,<2.7",  # Installing torch 2.7 results in slower compiled LLMs. Pinned while we investigate.
+    "torch>=2.1",
     "torchaudio",
     "torchvision",
     "pyctcdecode>=0.4.0",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 70cb07440f..e75872d479 100644
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -91,7 +91,7 @@ deps = {
     "tiktoken": "tiktoken",
     "timm": "timm<=1.0.11",
     "tokenizers": "tokenizers>=0.21,<0.22",
-    "torch": "torch>=2.1,<2.7",
+    "torch": "torch>=2.1",
     "torchaudio": "torchaudio",
     "torchvision": "torchvision",
     "pyctcdecode": "pyctcdecode>=0.4.0",