SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[deps] pin max torch version (#37760)

Browse Source 
pin max pt version :(
This commit is contained in:
Joao Gante
2025-04-24 16:18:25 +01:00
committed by GitHub
parent 0302aa1c6e
commit af6d2756d9
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -189,7 +189,7 @@ _deps = [
"tiktoken", "tiktoken",
"timm<=1.0.11", "timm<=1.0.11",
"tokenizers>=0.21,<0.22", "tokenizers>=0.21,<0.22",
"torch>=2.1", "torch>=2.1,<2.7", # Installing torch 2.7 results in slower compiled LLMs. Pinned while we investigate.
"torchaudio", "torchaudio",
"torchvision", "torchvision",
"pyctcdecode>=0.4.0", "pyctcdecode>=0.4.0",

2
src/transformers/dependency_versions_table.py
View File

@@ -91,7 +91,7 @@ deps = {
"tiktoken": "tiktoken", "tiktoken": "tiktoken",
"timm": "timm<=1.0.11", "timm": "timm<=1.0.11",
"tokenizers": "tokenizers>=0.21,<0.22", "tokenizers": "tokenizers>=0.21,<0.22",
"torch": "torch>=2.1", "torch": "torch>=2.1,<2.7",
"torchaudio": "torchaudio", "torchaudio": "torchaudio",
"torchvision": "torchvision", "torchvision": "torchvision",
"pyctcdecode": "pyctcdecode>=0.4.0", "pyctcdecode": "pyctcdecode>=0.4.0",