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

Pin PyTorch while we fix compatibility with 1.12

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

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

2
src/transformers/dependency_versions_table.py
View File

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