From 0f502682fb08c7dac5b255903f63dcd9cc2d68eb Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <sylvain.gugger@gmail.com>
Date: Thu, 21 Oct 2021 11:59:23 -0400
Subject: [PATCH] Pin PyTorch to make CI green

---
 setup.py                                      | 2 +-
 src/transformers/dependency_versions_table.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 11f1ac7e8bb..e5d1fae126d 100644
--- 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",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 9168ad5c05c..ee153430249 100644
--- 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",