From c6fa514b4ce2352a006d4d4e3864f10cd18febd6 Mon Sep 17 00:00:00 2001
From: Stanislas0 <qinkai.zheng1028@gmail.com>
Date: Mon, 13 Mar 2023 15:14:13 +0800
Subject: [PATCH] Update requirements

---
 requirements.txt | 2 +-
 setup.py         | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 75c8dd4..03916f0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,4 @@ transformers>=4.22.0
 torch>=1.10.0
 tqdm>=4.63.0
 cpm_kernels
-deepspeed
+deepspeed>0.6.1
\ No newline at end of file
diff --git a/setup.py b/setup.py
index af193f7..3260672 100644
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,8 @@ setup(
         "tokenizers>=0.11.0,<=0.11.4",
         "torch>=1.10.0",
         "tqdm>=4.63.0",
+        "cpm_kernels",
+        "deepspeed>0.6.1",
     ],
     entry_points={}
 )