From a1faf22f2ce5b96b16f87d26e1243d4b14d5ca56 Mon Sep 17 00:00:00 2001 From: Yoach Lacombe <52246514+ylacombe@users.noreply.github.com> Date: Wed, 4 Sep 2024 18:28:08 +0200 Subject: [PATCH] [BUG] fix upper nltk version (#33301) fix upper nltk version --- 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 26cb0faca0713e..2b736d5ddd3890 100644 --- a/setup.py +++ b/setup.py @@ -130,7 +130,7 @@ "keras>2.9,<2.16", "keras-nlp>=0.3.1,<0.14.0", # keras-nlp 0.14 doesn't support keras 2, see pin on keras. "librosa", - "nltk", + "nltk<=3.8.1", "natten>=0.14.6,<0.15.0", "numpy>=1.17", "onnxconverter-common", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index d8347371b3ad89..83fb6797f83d44 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -36,7 +36,7 @@ "keras": "keras>2.9,<2.16", "keras-nlp": "keras-nlp>=0.3.1,<0.14.0", "librosa": "librosa", - "nltk": "nltk", + "nltk": "nltk<=3.8.1", "natten": "natten>=0.14.6,<0.15.0", "numpy": "numpy>=1.17", "onnxconverter-common": "onnxconverter-common",