From 3345ae733b6f4aeb7204a0f3e646a3cdbaad0023 Mon Sep 17 00:00:00 2001
From: Joao Gante <joaofranciscocardosogante@gmail.com>
Date: Mon, 1 Jul 2024 17:39:33 +0100
Subject: [PATCH] dependencies: `keras-nlp<0.14` pin (#31684)

* keras nlp pin

* this should use the new docker images:dev

* dev-ci
---
 setup.py                                      | 2 +-
 src/transformers/dependency_versions_table.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 8f28f0f9e4..af2cec8aa9 100644
--- a/setup.py
+++ b/setup.py
@@ -128,7 +128,7 @@ _deps = [
     "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
index 7f8c3e4433..fcb2c7a29f 100644
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -34,7 +34,7 @@ deps = {
     "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",