Also pin TensorFlow CPU

37e01633 · Sylvain Gugger · a3f74580 · 37e01633 · 37e01633
Unverified Commit 37e01633 authored Nov 18, 2022 by Sylvain Gugger
Show 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
@@ -156,7 +156,7 @@ _deps = [
    "sigopt",
    "librosa",
    "starlette",
-    "tensorflow-cpu>=2.3",
+    "tensorflow-cpu>=2.4,<2.11",
    "tensorflow>=2.4,<2.11",
    "tensorflow-text",
    "tf2onnx",

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