diff --git a/setup.py b/setup.py
index 0799b5ad0c..16ae3d92a8 100644
--- a/setup.py
+++ b/setup.py
@@ -146,7 +146,7 @@ def _parse_requirements_file(file_path):
 _transformers_integration_deps = [
     f"{'nm-transformers' if is_release else 'nm-transformers-nightly'}"
     f"~={version_base}",
-    "datasets<=1.18.4",
+    "datasets<=2.11",
     "scikit-learn",
     "seqeval",
 ]