From 62396cff46854dc53023236cfeb785993fa70067 Mon Sep 17 00:00:00 2001 From: Matt Date: Tue, 22 Aug 2023 13:13:38 +0100 Subject: [PATCH] TF 2.14 compatibility (#25630) * Update the TF pin and see if anything breaks * make fixup * make fixup * make fixup --- examples/tensorflow/_tests_requirements.txt | 2 +- setup.py | 6 +++--- src/transformers/dependency_versions_table.py | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/examples/tensorflow/_tests_requirements.txt b/examples/tensorflow/_tests_requirements.txt index e28367c01c..989bda3302 100644 --- a/examples/tensorflow/_tests_requirements.txt +++ b/examples/tensorflow/_tests_requirements.txt @@ -1,4 +1,4 @@ -tensorflow<2.14 +tensorflow<2.15 tensorboard scikit-learn seqeval diff --git a/setup.py b/setup.py index 3de839dcd5..b95ec018f5 100644 --- a/setup.py +++ b/setup.py @@ -166,9 +166,9 @@ _deps = [ "sudachipy>=0.6.6", "sudachidict_core>=20220729", # TensorFlow pin. When changing this value, update examples/tensorflow/_tests_requirements.txt accordingly - "tensorflow-cpu>=2.6,<2.14", - "tensorflow>=2.6,<2.14", - "tensorflow-text<2.14", + "tensorflow-cpu>=2.6,<2.15", + "tensorflow>=2.6,<2.15", + "tensorflow-text<2.15", "tf2onnx", "timeout-decorator", "timm", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 117180c2ee..0bfa526c2b 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -72,9 +72,9 @@ deps = { "starlette": "starlette", "sudachipy": "sudachipy>=0.6.6", "sudachidict_core": "sudachidict_core>=20220729", - "tensorflow-cpu": "tensorflow-cpu>=2.6,<2.14", - "tensorflow": "tensorflow>=2.6,<2.14", - "tensorflow-text": "tensorflow-text<2.14", + "tensorflow-cpu": "tensorflow-cpu>=2.6,<2.15", + "tensorflow": "tensorflow>=2.6,<2.15", + "tensorflow-text": "tensorflow-text<2.15", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator", "timm": "timm",