dependencies: keras-nlp<0.14 pin (huggingface#31684)

* keras nlp pin * this should use the new docker images:dev * dev-ci
fxmarty · Jul 1, 2024 · 3345ae7 · 3345ae7
1 parent e655029
commit 3345ae7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -128,7 +128,7 @@
     "kenlm",
     # Keras pin - this is to make sure Keras 3 doesn't destroy us. Remove or change when we have proper support.
     "keras>2.9,<2.16",
-    "keras-nlp>=0.3.1",
+    "keras-nlp>=0.3.1,<0.14.0",  # keras-nlp 0.14 doesn't support keras 2, see pin on keras.
     "librosa",
     "nltk",
     "natten>=0.14.6,<0.15.0",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -34,7 +34,7 @@
     "jinja2": "jinja2>=3.1.0",
     "kenlm": "kenlm",
     "keras": "keras>2.9,<2.16",
-    "keras-nlp": "keras-nlp>=0.3.1",
+    "keras-nlp": "keras-nlp>=0.3.1,<0.14.0",
     "librosa": "librosa",
     "nltk": "nltk",
     "natten": "natten>=0.14.6,<0.15.0",