Pin tensorflow-text to go with tensorflow (#22362)

* Pin tensorflow-text to go with tensorflow * Make it more convenient to pin TensorFlow * setup don't like f-strings
huggingface · Mar 24, 2023 · 6587125 · 6587125
1 parent 0120347
commit 6587125
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -164,9 +164,10 @@
     "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
@@ -72,7 +72,7 @@
     "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",