diff --git a/requirements.txt b/requirements.txt
index 952f906..9f49de2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,4 @@ torch>=2.2,<2.6
 numpy>=1.23,<3
 tqdm>=4.70.1,<5
 zstandard>=0.22,<0.24
-datasets>=2.19,<3
+datasets>=5.1.0,<6