diff --git a/README.md b/README.md index 2c73467..9e072a0 100644 --- a/README.md +++ b/README.md @@ -32,7 +32,7 @@ ChatGLM-6B 使用了和 ChatGPT 相似的技术,针对中文问答和对话进 ### 环境安装 -使用 pip 安装依赖:`pip install -r requirements.txt`,其中 `transformers` 库版本推荐为 `4.26.1`,但理论上不低于 `4.23.1` 即可。 +使用 pip 安装依赖:`pip install -r requirements.txt`,其中 `transformers` 库版本推荐为 `4.27.1`,但理论上不低于 `4.23.1` 即可。 ### 代码调用 diff --git a/README_en.md b/README_en.md index b5b4b62..d5c05bb 100644 --- a/README_en.md +++ b/README_en.md @@ -32,7 +32,7 @@ If you have other good projects, please refer to the above format to add to READ ### Environment Setup -Install the requirements with pip: `pip install -r requirements.txt`. `transformers` library version is recommended to be `4.26.1`, but theoretically any version no lower than `4.23.1` is acceptable. +Install the requirements with pip: `pip install -r requirements.txt`. `transformers` library version is recommended to be `4.27.1`, but theoretically any version no lower than `4.23.1` is acceptable. ### Usage diff --git a/ptuning/README.md b/ptuning/README.md index a4097f6..4434e3a 100644 --- a/ptuning/README.md +++ b/ptuning/README.md @@ -4,7 +4,7 @@ 下面以 [ADGEN](https://aclanthology.org/D19-1321.pdf) (广告生成) 数据集为例介绍代码的使用方法。 ## 软件依赖 -除 ChatGLM-6B 的依赖之外,还需要按照以下依赖 +运行微调需要4.27.1版本的`transformers`。除 ChatGLM-6B 的依赖之外,还需要按照以下依赖 ``` pip install rouge_chinese nltk jieba datasets ``` diff --git a/requirements.txt b/requirements.txt index 2948480..00707fe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ protobuf>=3.19.5,<3.20.1 -transformers==4.26.1 +transformers==4.27.1 icetk cpm_kernels torch>=1.10