add accelerate, cpm_kernels 依赖

2023-08-20 14:29:37 +08:00 · 2023-08-20 14:29:37 +08:00 · a8e7fb4ac2
parent 80602dcae1
commit a8e7fb4ac2
1 changed files with 1 additions and 1 deletions
--- a/ptuning/README.md
+++ b/ptuning/README.md
@ -6,7 +6,7 @@
 ## 软件依赖
 运行微调除 ChatGLM2-6B 的依赖之外，还需要安装以下依赖
 ```
-pip install rouge_chinese nltk jieba datasets
+pip install rouge_chinese nltk jieba datasets accelerate cpm_kernels
 ```
 ## 使用方法