diff --git a/pyproject.toml b/pyproject.toml
index 526916b..8223d6d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ core = [
     "accelerate==1.6.0",
     "transformers[sentencepiece,torch-vision]==4.53.0",
     "tensorboard==2.18.0",
-    "datasets==3.1.0",
+    "datasets==5.0.0",
 ]
 
 gpu = [