From a3f745806662dd2a36aa31e4e3d0da888f2f58f3 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Fri, 18 Nov 2022 07:12:55 -0500 Subject: [PATCH] Pin to the right version... --- examples/tensorflow/_tests_requirements.txt | 2 +- setup.py | 2 +- src/transformers/dependency_versions_table.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/tensorflow/_tests_requirements.txt b/examples/tensorflow/_tests_requirements.txt index 37e37e352..837ce6d0d 100644 --- a/examples/tensorflow/_tests_requirements.txt +++ b/examples/tensorflow/_tests_requirements.txt @@ -1,4 +1,4 @@ -tensorflow +tensorflow<2.11 tensorboard scikit-learn seqeval diff --git a/setup.py b/setup.py index 0b1d93a46..31f24f567 100644 --- a/setup.py +++ b/setup.py @@ -157,7 +157,7 @@ _deps = [ "librosa", "starlette", "tensorflow-cpu>=2.3", - "tensorflow>=2.4,<2.10", + "tensorflow>=2.4,<2.11", "tensorflow-text", "tf2onnx", "timeout-decorator", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 6946945cd..07d3add4c 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,<2.10", + "tensorflow": "tensorflow>=2.4,<2.11", "tensorflow-text": "tensorflow-text", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator",