Pin PyTorch to make CI green

0f502682 · Sylvain Gugger · f9c16b02 · 0f502682 · 0f502682
Commit 0f502682 authored Oct 21, 2021 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
@@ -149,7 +149,7 @@ _deps = [
    "timeout-decorator",
    "timm",
    "tokenizers>=0.10.1,<0.11",
-    "torch>=1.0",
+    "torch>=1.0,<1.10",
    "torchaudio",
    "tqdm>=4.27",
    "unidic>=1.0.2",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -67,7 +67,7 @@ deps = {
    "timeout-decorator": "timeout-decorator",
    "timm": "timm",
    "tokenizers": "tokenizers>=0.10.1,<0.11",
-    "torch": "torch>=1.0",
+    "torch": "torch>=1.0,<1.10",
    "torchaudio": "torchaudio",
    "tqdm": "tqdm>=4.27",
    "unidic": "unidic>=1.0.2",