riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Pin torchvision and torchaudio as well

Browse Source
This commit is contained in:
Matt 2024-01-30 18:14:20 +00:00
parent fd747a3c33
commit 05fa5e0bab
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -176,8 +176,8 @@ _deps = [
"timm", "timm",
"tokenizers>=0.14,<0.19", "tokenizers>=0.14,<0.19",
"torch>=1.11,!=1.12.0,<2.2.0", "torch>=1.11,!=1.12.0,<2.2.0",
"torchaudio", "torchaudio<2.2.0",
"torchvision", "torchvision<0.17.0",
"pyctcdecode>=0.4.0", "pyctcdecode>=0.4.0",
"tqdm>=4.27", "tqdm>=4.27",
"unidic>=1.0.2", "unidic>=1.0.2",

4
src/transformers/dependency_versions_table.py
View File

@ -81,8 +81,8 @@ deps = {
"timm": "timm", "timm": "timm",
"tokenizers": "tokenizers>=0.14,<0.19", "tokenizers": "tokenizers>=0.14,<0.19",
"torch": "torch>=1.11,!=1.12.0,<2.2.0", "torch": "torch>=1.11,!=1.12.0,<2.2.0",
"torchaudio": "torchaudio", "torchaudio": "torchaudio<2.2.0",
"torchvision": "torchvision", "torchvision": "torchvision<0.17.0",
"pyctcdecode": "pyctcdecode>=0.4.0", "pyctcdecode": "pyctcdecode>=0.4.0",
"tqdm": "tqdm>=4.27", "tqdm": "tqdm>=4.27",
"unidic": "unidic>=1.0.2", "unidic": "unidic>=1.0.2",