From 41b1e41662bfea78c58e479e822cd60d938c4e26 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Thu, 17 Oct 2024 22:11:12 +0000
Subject: [PATCH] Bump deepspeed from 0.14.4 to 0.15.1

Bumps [deepspeed](https://github.com/microsoft/DeepSpeed) from 0.14.4 to 0.15.1.
- [Release notes](https://github.com/microsoft/DeepSpeed/releases)
- [Commits](https://github.com/microsoft/DeepSpeed/compare/v0.14.4...v0.15.1)

---
updated-dependencies:
- dependency-name: deepspeed
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 3f17e5d329..ab1ca76ec2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ tokenizers>=0.19.1
 bitsandbytes==0.44.0
 accelerate==0.34.2
 datasets==2.21.0
-deepspeed==0.14.4
+deepspeed==0.15.1
 pydantic==2.6.3
 addict
 fire
diff --git a/setup.py b/setup.py
index 1b64fadaef..6f7d86f360 100644
--- a/setup.py
+++ b/setup.py
@@ -86,7 +86,7 @@ def parse_requirements():
             "fused-dense-lib  @ git+https://github.com/Dao-AILab/flash-attention@v2.6.2#subdirectory=csrc/fused_dense_lib",
         ],
         "deepspeed": [
-            "deepspeed==0.14.4",
+            "deepspeed==0.15.1",
             "deepspeed-kernels",
         ],
         "mamba-ssm": [