Bump torch requirement to >= 2 (#35479)

Bump torch requirement, follow-up of #35358
2025-01-08 22:59:32 +08:00
parent 59e5b3f01b
commit 1a6c1d3a9a
2 changed files with 2 additions and 2 deletions
--- 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",
--- 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",