From 23fe46f74afe46052d57843a00afe70235b87278 Mon Sep 17 00:00:00 2001 From: SimonLee Date: Mon, 5 Feb 2024 16:17:35 +0800 Subject: [PATCH] update ptuning readme for pip install --- 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 f92a328..69412a4 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 +pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece ``` ## 使用方法 diff --git a/ptuning/README_en.md b/ptuning/README_en.md index 34a68a6..1a5fa3c 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 +pip install rouge_chinese nltk jieba datasets cpm_kernels sentencepiece ``` ## Instructions