Pin PyTorch while we fix compatibility with 1.12

5a3d0cbd · Sylvain Gugger · 6c8f4c9a · 5a3d0cbd · 5a3d0cbd
Commit 5a3d0cbd authored Jun 28, 2022 by Sylvain Gugger
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +1 -1

src/transformers/dependency_versions_table.py src/transformers/dependency_versions_table.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -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",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -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",