From a200b7628f8d5ce038b26c2576384a12c9e0a795 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Wed, 23 Oct 2024 20:38:27 +0800
Subject: [PATCH] chore: set deepspeed version

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 412caa50..d5b4b8cf 100644
--- a/setup.py
+++ b/setup.py
@@ -37,7 +37,7 @@ extra_require = {
     "torch": ["torch>=1.13.1"],
     "torch-npu": ["torch==2.1.0", "torch-npu==2.1.0.post3", "decorator"],
     "metrics": ["nltk", "jieba", "rouge-chinese"],
-    "deepspeed": ["deepspeed>=0.10.0"],
+    "deepspeed": ["deepspeed==0.15.2"],
     "liger-kernel": ["liger-kernel"],
     "bitsandbytes": ["bitsandbytes>=0.39.0"],
     "hqq": ["hqq"],