Skip to content

Commit

Permalink
deps: increases deepspeed's lower-bound to >=1.15.1
Browse files Browse the repository at this point in the history
force inclusion of recent security patch

Signed-off-by: James Kunstle <[email protected]>
  • Loading branch information
JamesKunstle committed Nov 12, 2024
1 parent 2d6173b commit 51dea40
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
4 changes: 0 additions & 4 deletions requirements-cuda.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
5 changes: 5 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 51dea40

Please sign in to comment.