diff --git a/setup.py b/setup.py index 43e70cdf..4fa7a721 100644 --- a/setup.py +++ b/setup.py @@ -49,7 +49,7 @@ setuptools.setup( 'numpy>=1.14.5', 'torch>=1.9.0,<1.14.0', 'transformers==4.22.2', - 'datasets==2.9.0', + 'datasets==2.11.0', 'pathos==0.2.8', 'protobuf==3.20.1', # for sts: