Also pin TensorFlow CPU

2022-11-18 08:50:56 -05:00
parent a3f7458066
commit 37e016331f
2 changed files with 2 additions and 2 deletions
--- 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",