diff --git a/requirements.txt b/requirements.txt
index ed73c0c..75c8dd4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,5 +6,7 @@ pyzmq>=23.2.1
 regex>=2022.3.15
 setuptools>=58.0.4
 transformers>=4.22.0
-pytorch>=1.10.0
-tqdm>=4.63.0
\ No newline at end of file
+torch>=1.10.0
+tqdm>=4.63.0
+cpm_kernels
+deepspeed