diff --git a/setup.py b/setup.py index 27ab6efd69..536ece1c8e 100644 --- a/setup.py +++ b/setup.py @@ -102,7 +102,7 @@ _deps = [ "codecarbon==1.2.0", "cookiecutter==1.7.3", "dataclasses", - "datasets", + "datasets!=2.5.0", "deepspeed>=0.6.5", "dill<0.3.5", "evaluate>=0.2.0", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 74c6d00c2a..e34b615edb 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -8,7 +8,7 @@ deps = { "codecarbon": "codecarbon==1.2.0", "cookiecutter": "cookiecutter==1.7.3", "dataclasses": "dataclasses", - "datasets": "datasets", + "datasets": "datasets!=2.5.0", "deepspeed": "deepspeed>=0.6.5", "dill": "dill<0.3.5", "evaluate": "evaluate>=0.2.0",