Pin TF until tests are fixed (#14283)

* Pin TF until tests are fixed * Also pin TF CPU

Pin TF until tests are fixed (#14283)
* Pin TF until tests are fixed * Also pin TF CPU
a14d62b0 · Sylvain Gugger · GitHub · b90a48f6 · a14d62b0 · a14d62b0
Unverified Commit a14d62b0 authored Nov 04, 2021 by Sylvain Gugger Committed by GitHub Nov 04, 2021
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

setup.py setup.py +2 -2

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

No files found.
--- 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",

--- 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",