Pin TensorFlow to the next release (#20635)

6c1a0b39 · Sylvain Gugger · GitHub · c95f8470 · 6c1a0b39 · 6c1a0b39
Unverified Commit 6c1a0b39 authored Dec 06, 2022 by Sylvain Gugger Committed by GitHub Dec 06, 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
@@ -158,8 +158,8 @@ _deps = [
    "sigopt",
    "librosa",
    "starlette",
-    "tensorflow-cpu>=2.4,<2.11",
-    "tensorflow>=2.4,<2.11",
+    "tensorflow-cpu>=2.4,<2.12",
+    "tensorflow>=2.4,<2.12",
    "tensorflow-text",
    "tf2onnx",
    "timeout-decorator",

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