diff --git a/setup.py b/setup.py index 04942c88b43..d47ccb197c4 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 70cb07440fb..e75872d4790 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",