diff --git a/setup.py b/setup.py index 4fa848590..114642902 100644 --- a/setup.py +++ b/setup.py @@ -57,7 +57,7 @@ "click>=7.1.2,!=8.0.0", # 8.0.0 blocked due to reported bug "torch>=1.7.0", "transformers>4.0,<5.0", - "datasets<2.19", + "datasets", "accelerate>=0.20.3", "compressed-tensors" if version_info.is_release