Pin TensorFlow (huggingface#20313)

mpierrau · Dec 15, 2022 · 90e95e1 · 90e95e1
1 parent baca2c1
commit 90e95e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -157,7 +157,7 @@
  "librosa",
  "starlette",
  "tensorflow-cpu>=2.3",
- "tensorflow>=2.4",
+ "tensorflow>=2.4,<2.10",
  "tensorflow-text",
  "tf2onnx",
  "timeout-decorator",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -63,7 +63,7 @@
  "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",