diff --git a/setup.py b/setup.py index 047ae7f0faa..24ac227701d 100644 --- a/setup.py +++ b/setup.py @@ -79,7 +79,7 @@ _transformers_deps = _pytorch_deps + [ f"{'nm-transformers' if is_release else 'nm-transformers-nightly'}" f"~={version_nm_deps}", - "datasets<=2.11", + "datasets<=2.14.6", "scikit-learn", "seqeval", "einops",