From 79e210dad34ae31a647a023eda86cd807a3ba7c4 Mon Sep 17 00:00:00 2001 From: SimonLee Date: Sun, 18 Feb 2024 11:28:35 +0800 Subject: [PATCH] missing accelerate by deepspeed --- ptuning/README.md | 2 +- ptuning/README_en.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ptuning/README.md b/ptuning/README.md index 69412a4..d833d1d 100644 --- a/ptuning/README.md +++ b/ptuning/README.md @@ -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 ``` ## 使用方法 diff --git a/ptuning/README_en.md b/ptuning/README_en.md index 1a5fa3c..7d0b2b1 100644 --- a/ptuning/README_en.md +++ b/ptuning/README_en.md @@ -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