diff --git a/examples/tensorflow/_tests_requirements.txt b/examples/tensorflow/_tests_requirements.txt index 837ce6d0d16..4548479bcc9 100644 --- a/examples/tensorflow/_tests_requirements.txt +++ b/examples/tensorflow/_tests_requirements.txt @@ -1,4 +1,4 @@ -tensorflow<2.11 +tensorflow<2.12 tensorboard scikit-learn seqeval diff --git a/setup.py b/setup.py index 669c8d959ca..803535ab53d 100644 --- a/setup.py +++ b/setup.py @@ -164,9 +164,10 @@ _deps = [ "starlette", "sudachipy>=0.6.6", "sudachidict_core>=20220729", + # TensorFlow pin. When changing this value, update examples/tensorflow/_tests_requirements.txt accordingly "tensorflow-cpu>=2.4,<2.12", "tensorflow>=2.4,<2.12", - "tensorflow-text", + "tensorflow-text<2.12", "tf2onnx", "timeout-decorator", "timm", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 7c596c26683..1fbddb013fb 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -72,7 +72,7 @@ deps = { "sudachidict_core": "sudachidict_core>=20220729", "tensorflow-cpu": "tensorflow-cpu>=2.4,<2.12", "tensorflow": "tensorflow>=2.4,<2.12", - "tensorflow-text": "tensorflow-text", + "tensorflow-text": "tensorflow-text<2.12", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator", "timm": "timm",