diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 37decda8..6230797e 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -3,7 +3,3 @@ bitsandbytes>=0.43.1 # required for FSDP updates accelerate>=0.34.2 - -# deepspeed needs to be at the end or it'll break stuff. -# only required for cuda, not intended for ROCm or Gaudi -deepspeed>=0.14.3 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 60ba5599..ca84c8bc 100644 --- a/requirements.txt +++ b/requirements.txt @@ -22,3 +22,8 @@ peft pydantic>=2.7.0 aiofiles>=23.2.1 + +# deepspeed needs to be at the end or it'll break stuff. +# only required for cuda, not intended for ROCm or Gaudi, but installation +# required for symbol resolution. +deepspeed>=0.15.1 \ No newline at end of file