diff --git a/setup.py b/setup.py index 8f28f0f9e4adb7..af2cec8aa9d5e3 100644 --- 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 index 7f8c3e4433fe3c..fcb2c7a29f63bb 100644 --- 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",