riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-01 20:00:09 +06:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

build: 📌 Remove upper bound on PyTorch (#38789)

Browse Source 
build: 📌 remove upper bound on torch dependency as issue which originally resulted in the pin has been released in torch 2.7.1
This commit is contained in:
Kyle Mylonakis 2025-06-12 15:34:13 +01:00 committed by GitHub
parent eea35a15b0
commit 3542e0b844
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -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",

2
src/transformers/dependency_versions_table.py
View File

@ -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",