dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Pin PyTorch while we fix compatibility with 1.12

This commit is contained in:
Sylvain Gugger 2022-06-28 15:07:26 -04:00
parent 6c8f4c9a93
commit 5a3d0cbdda
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

@ -160,7 +160,7 @@ _deps = [
"timeout-decorator",
"timm",
"tokenizers>=0.11.1,!=0.11.3,<0.13",
"torch>=1.0",
"torch>=1.0,<1.12",
"torchaudio",
"pyctcdecode>=0.3.0",
"tqdm>=4.27",

2
src/transformers/dependency_versions_table.py
View File

@ -66,7 +66,7 @@ deps = {
"timeout-decorator": "timeout-decorator",
"timm": "timm",
"tokenizers": "tokenizers>=0.11.1,!=0.11.3,<0.13",
"torch": "torch>=1.0",
"torch": "torch>=1.0,<1.12",
"torchaudio": "torchaudio",
"pyctcdecode": "pyctcdecode>=0.3.0",
"tqdm": "tqdm>=4.27",