From fea4636cfaccb27fabc0c1d742e7667b17fcc8cd Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Wed, 31 Aug 2022 09:07:53 +0100 Subject: [PATCH] Pin max tf version (#18818) --- setup.py | 4 ++-- src/transformers/dependency_versions_table.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 05ec2c7617..e6f442e5ec 100644 --- a/setup.py +++ b/setup.py @@ -154,8 +154,8 @@ _deps = [ "sigopt", "librosa", "starlette", - "tensorflow-cpu>=2.3", - "tensorflow>=2.3", + "tensorflow-cpu>=2.3,<2.10", + "tensorflow>=2.3,<2.10", "tensorflow-text", "tf2onnx", "timeout-decorator", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index be3dba684b..fec2fc9203 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -60,8 +60,8 @@ deps = { "sigopt": "sigopt", "librosa": "librosa", "starlette": "starlette", - "tensorflow-cpu": "tensorflow-cpu>=2.3", - "tensorflow": "tensorflow>=2.3", + "tensorflow-cpu": "tensorflow-cpu>=2.3,<2.10", + "tensorflow": "tensorflow>=2.3,<2.10", "tensorflow-text": "tensorflow-text", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator",