From 4989e73e2f86f991a7c2aec4852d873903c71c14 Mon Sep 17 00:00:00 2001 From: Matt Date: Thu, 16 Nov 2023 13:47:43 +0000 Subject: [PATCH] Update the TF pin for 2.15 (#27375) * Move the TF pin for 2.15 * make fixup --- examples/tensorflow/_tests_requirements.txt | 4 ++-- setup.py | 8 ++++---- src/transformers/dependency_versions_table.py | 8 ++++---- 3 files changed, 10 insertions(+), 10 deletions(-) diff --git a/examples/tensorflow/_tests_requirements.txt b/examples/tensorflow/_tests_requirements.txt index e9351b2b42..161a045bd5 100644 --- a/examples/tensorflow/_tests_requirements.txt +++ b/examples/tensorflow/_tests_requirements.txt @@ -1,5 +1,5 @@ -tensorflow<2.15 -keras<2.15 +tensorflow<2.16 +keras<2.16 tensorboard scikit-learn seqeval diff --git a/setup.py b/setup.py index e42a29585f..764411dab3 100644 --- a/setup.py +++ b/setup.py @@ -127,7 +127,7 @@ _deps = [ "jieba", "kenlm", # Keras pin - this is to make sure Keras 3 doesn't destroy us. Remove or change when we have proper support. - "keras<2.15", + "keras<2.16", "keras-nlp>=0.3.1", "librosa", "nltk", @@ -169,9 +169,9 @@ _deps = [ "sudachidict_core>=20220729", "tensorboard", # TensorFlow pin. When changing this value, update examples/tensorflow/_tests_requirements.txt accordingly - "tensorflow-cpu>=2.6,<2.15", - "tensorflow>=2.6,<2.15", - "tensorflow-text<2.15", + "tensorflow-cpu>=2.6,<2.16", + "tensorflow>=2.6,<2.16", + "tensorflow-text<2.16", "tf2onnx", "timeout-decorator", "timm", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 736b312810..d460790c8e 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -33,7 +33,7 @@ deps = { "jaxlib": "jaxlib>=0.4.1,<=0.4.13", "jieba": "jieba", "kenlm": "kenlm", - "keras": "keras<2.15", + "keras": "keras<2.16", "keras-nlp": "keras-nlp>=0.3.1", "librosa": "librosa", "nltk": "nltk", @@ -74,9 +74,9 @@ deps = { "sudachipy": "sudachipy>=0.6.6", "sudachidict_core": "sudachidict_core>=20220729", "tensorboard": "tensorboard", - "tensorflow-cpu": "tensorflow-cpu>=2.6,<2.15", - "tensorflow": "tensorflow>=2.6,<2.15", - "tensorflow-text": "tensorflow-text<2.15", + "tensorflow-cpu": "tensorflow-cpu>=2.6,<2.16", + "tensorflow": "tensorflow>=2.6,<2.16", + "tensorflow-text": "tensorflow-text<2.16", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator", "timm": "timm",