From f7ab8c425124eb40e288d05363f2228bc9606efb Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Fri, 18 Nov 2022 06:57:15 -0500 Subject: [PATCH] Pin TensorFlow (#20313) --- setup.py | 2 +- src/transformers/dependency_versions_table.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 2f284e3bb..0b1d93a46 100644 --- 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", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 1d6223f2a..6946945cd 100644 --- 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",