Skip to content

GitLab

Commit 5a3d0cbd authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Pin PyTorch while we fix compatibility with 1.12

parent 6c8f4c9a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ 5a3d0cbd
...@@ -160,7 +160,7 @@ _deps = [ ...@@ -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",
......
src/transformers/dependency_versions_table.py
View file @ 5a3d0cbd
...@@ -66,7 +66,7 @@ deps = { ...@@ -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",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment