diff --git a/cmd/runtimes/deepspeed/requirements.txt b/cmd/runtimes/deepspeed/requirements.txt index 3efc253dfd..eb4b622240 100644 --- a/cmd/runtimes/deepspeed/requirements.txt +++ b/cmd/runtimes/deepspeed/requirements.txt @@ -1,7 +1,7 @@ # Keep the same version as for Torch runtime. torch==2.7.1 # DeepSpeed libraries. -deepspeed==0.17.4 +deepspeed==0.18.4 mpi4py==4.1.0 # HuggingFace libraries. datasets==4.0.0