From 1a6c1d3a9a3eaf459062943b05866346549ae8a9 Mon Sep 17 00:00:00 2001
From: Yichen Yan <oraluben@outlook.com>
Date: Wed, 8 Jan 2025 22:59:32 +0800
Subject: [PATCH] Bump torch requirement to >= 2 (#35479)

Bump torch requirement, follow-up of #35358
---
 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 a78bb20dd0..1a4d94c24e 100644
--- a/setup.py
+++ b/setup.py
@@ -182,7 +182,7 @@ _deps = [
     "tiktoken",
     "timm<=1.0.11",
     "tokenizers>=0.21,<0.22",
-    "torch",
+    "torch>=2.0",
     "torchaudio",
     "torchvision",
     "pyctcdecode>=0.4.0",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 6a737b805a..26500c22b1 100644
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -87,7 +87,7 @@ deps = {
     "tiktoken": "tiktoken",
     "timm": "timm<=1.0.11",
     "tokenizers": "tokenizers>=0.21,<0.22",
-    "torch": "torch",
+    "torch": "torch>=2.0",
     "torchaudio": "torchaudio",
     "torchvision": "torchvision",
     "pyctcdecode": "pyctcdecode>=0.4.0",