github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

missing accelerate by deepspeed

pull/1455/head
SimonLee 2024-02-18 11:28:35 +08:00
parent 0b7dc01921
commit 79e210dad3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ptuning/README.md
View File

@ -8,7 +8,7 @@
## 软件依赖
运行微调需要4.27.1版本的`transformers`。除 ChatGLM-6B 的依赖之外，还需要安装以下依赖
```
pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece
pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece accelerate
```
## 使用方法

2
ptuning/README_en.md
View File

@ -6,7 +6,7 @@ The following uses the [ADGEN](https://aclanthology.org/D19-1321.pdf) (advertisi
## Software dependencies
Running p-tuning requires version 4.27.1 of `transformers`. In addition to the dependencies of ChatGLM-6B, the following dependencies are required
```
pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece
pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece accelerate
```
## Instructions