diff --git a/pyproject.toml b/pyproject.toml
index e6bb20a..46b8873 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = []
 [project.optional-dependencies]
 core = [
     "accelerate==1.6.0",
-    "datasets==5.0.1",
+    "datasets==5.1.0",
     "hydra-colorlog==1.2.0",
     "hydra-core==1.3.7",
     "tensorboard==2.18.0",