riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Bump torch requirement to >= 2 (#35479)

Browse Source 
Bump torch requirement, follow-up of #35358
This commit is contained in:
Yichen Yan 2025-01-08 22:59:32 +08:00 committed by GitHub
parent 59e5b3f01b
commit 1a6c1d3a9a
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

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

2
src/transformers/dependency_versions_table.py
View File

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