diff --git a/cmd/runtimes/deepspeed/requirements.txt b/cmd/runtimes/deepspeed/requirements.txt
index 3efc253dfd..da21b53a37 100644
--- a/cmd/runtimes/deepspeed/requirements.txt
+++ b/cmd/runtimes/deepspeed/requirements.txt
@@ -4,6 +4,6 @@ torch==2.7.1
 deepspeed==0.17.4
 mpi4py==4.1.0
 # HuggingFace libraries.
-datasets==4.0.0
+datasets==4.8.5
 transformers==4.55.0
 SentencePiece==0.2.0
diff --git a/cmd/runtimes/mlx/requirements.txt b/cmd/runtimes/mlx/requirements.txt
index 9eb8e59284..9c4a016c99 100644
--- a/cmd/runtimes/mlx/requirements.txt
+++ b/cmd/runtimes/mlx/requirements.txt
@@ -3,4 +3,4 @@ mlx[cuda]==0.28.0
 mlx-data==0.1.0
 mlx-lm==0.26.3
 # HuggingFace libraries.
-datasets==4.0.0
+datasets==4.8.5