Update requirements.txt

2 years ago · e16b10a068
parent f1afae7b26
commit e16b10a068
1 changed files with 4 additions and 2 deletions
--- 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
+torch>=1.10.0
 tqdm>=4.63.0
+cpm_kernels
+deepspeed