Pin TensorFlow (#20313)

f7ab8c42 · Sylvain Gugger · GitHub · b9826942 · f7ab8c42 · f7ab8c42
Unverified Commit f7ab8c42 authored Nov 18, 2022 by Sylvain Gugger Committed by GitHub Nov 18, 2022
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
@@ -157,7 +157,7 @@ _deps = [
    "librosa",
    "starlette",
    "tensorflow-cpu>=2.3",
-    "tensorflow>=2.4",
+    "tensorflow>=2.4,<2.10",
    "tensorflow-text",
    "tf2onnx",
    "timeout-decorator",

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