diff --git a/requirements.txt b/requirements.txt
index a5594f3..a12fae5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ numerize==0.12
 torchtyping==0.1.5
 rich==14.0.0
 accelerate==1.7.0
-datasets==3.6.0
+datasets==5.0.1
 sentencepiece==0.2.1
 matplotlib==3.10.3
 wandb==0.19.11