From 51dea402fdb1703352e653555a2e7670a87de2bb Mon Sep 17 00:00:00 2001 From: James Kunstle Date: Tue, 5 Nov 2024 13:38:25 -0800 Subject: [PATCH] deps: increases deepspeed's lower-bound to >=1.15.1 force inclusion of recent security patch Signed-off-by: James Kunstle --- requirements-cuda.txt | 4 ---- requirements.txt | 5 +++++ 2 files changed, 5 insertions(+), 4 deletions(-) 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