From a14d62b0b11e6cb0c64059606e4f8dbf78e40e41 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Thu, 4 Nov 2021 21:15:42 -0400
Subject: [PATCH] Pin TF until tests are fixed (#14283)

* Pin TF until tests are fixed

* Also pin TF CPU
---
 setup.py                                      | 4 ++--
 src/transformers/dependency_versions_table.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 0fb91a0cfc..f26d02efe8 100644
--- a/setup.py
+++ b/setup.py
@@ -151,8 +151,8 @@ _deps = [
     "sphinxext-opengraph==0.4.1",
     "sphinx-intl",
     "starlette",
-    "tensorflow-cpu>=2.3",
-    "tensorflow>=2.3",
+    "tensorflow-cpu>=2.3,<2.7",
+    "tensorflow>=2.3,<2.7",
     "timeout-decorator",
     "timm",
     "tokenizers>=0.10.1,<0.11",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 786e5a5691..b239280791 100644
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -62,8 +62,8 @@ deps = {
     "sphinxext-opengraph": "sphinxext-opengraph==0.4.1",
     "sphinx-intl": "sphinx-intl",
     "starlette": "starlette",
-    "tensorflow-cpu": "tensorflow-cpu>=2.3",
-    "tensorflow": "tensorflow>=2.3",
+    "tensorflow-cpu": "tensorflow-cpu>=2.3,<2.7",
+    "tensorflow": "tensorflow>=2.3,<2.7",
     "timeout-decorator": "timeout-decorator",
     "timm": "timm",
     "tokenizers": "tokenizers>=0.10.1,<0.11",