Pin max tf version (#18818)

fea4636c · Joao Gante · GitHub · 5c4c8690 · fea4636c · fea4636c
Unverified Commit fea4636c authored Aug 31, 2022 by Joao Gante Committed by GitHub Aug 31, 2022
Hide 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
@@ -154,8 +154,8 @@ _deps = [
    "sigopt",
    "librosa",
    "starlette",
-    "tensorflow-cpu>=2.3",
+    "tensorflow-cpu>=2.3,<2.10",
-    "tensorflow>=2.3",
+    "tensorflow>=2.3,<2.10",
    "tensorflow-text",
    "tf2onnx",
    "timeout-decorator",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -60,8 +60,8 @@ deps = {
    "sigopt": "sigopt",
    "librosa": "librosa",
    "starlette": "starlette",
-    "tensorflow-cpu": "tensorflow-cpu>=2.3",
+    "tensorflow-cpu": "tensorflow-cpu>=2.3,<2.10",
-    "tensorflow": "tensorflow>=2.3",
+    "tensorflow": "tensorflow>=2.3,<2.10",
    "tensorflow-text": "tensorflow-text",
    "tf2onnx": "tf2onnx",
    "timeout-decorator": "timeout-decorator",