Pin Keras for now (#26904)

* Pin Keras for now out of paranoia * Add the keras pin to _tests_requirements.txt too * Make sure the Keras version matches the TF one * make fixup
huggingface · Oct 19, 2023 · cbd278f · cbd278f
1 parent 73dc23f
commit cbd278f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/examples/tensorflow/_tests_requirements.txt b/examples/tensorflow/_tests_requirements.txt
@@ -1,4 +1,5 @@
 tensorflow<2.15
+keras<2.15
 tensorboard
 scikit-learn
 seqeval

diff --git a/setup.py b/setup.py
@@ -125,6 +125,8 @@
     "jaxlib>=0.4.1,<=0.4.13",
     "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-nlp>=0.3.1",
     "librosa",
     "nltk",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -32,6 +32,7 @@
     "jaxlib": "jaxlib>=0.4.1,<=0.4.13",
     "jieba": "jieba",
     "kenlm": "kenlm",
+    "keras": "keras<2.15",
     "keras-nlp": "keras-nlp>=0.3.1",
     "librosa": "librosa",
     "nltk": "nltk",