Skip to content

Commit af6d275

Browse files
authored
[deps] pin max torch version (#37760)
pin max pt version :(
1 parent 0302aa1 commit af6d275

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@
189189
"tiktoken",
190190
"timm<=1.0.11",
191191
"tokenizers>=0.21,<0.22",
192-
"torch>=2.1",
192+
"torch>=2.1,<2.7", # Installing torch 2.7 results in slower compiled LLMs. Pinned while we investigate.
193193
"torchaudio",
194194
"torchvision",
195195
"pyctcdecode>=0.4.0",

src/transformers/dependency_versions_table.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@
9191
"tiktoken": "tiktoken",
9292
"timm": "timm<=1.0.11",
9393
"tokenizers": "tokenizers>=0.21,<0.22",
94-
"torch": "torch>=2.1",
94+
"torch": "torch>=2.1,<2.7",
9595
"torchaudio": "torchaudio",
9696
"torchvision": "torchvision",
9797
"pyctcdecode": "pyctcdecode>=0.4.0",

0 commit comments

Comments
 (0)