Update requirements.txt

pull/313/head
duzx16 2023-03-31 15:18:21 +08:00
parent 24e24d5d6c
commit c206e7d9ad
4 changed files with 4 additions and 4 deletions

View File

@ -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` 即可。
### 代码调用 ### 代码调用

View File

@ -32,7 +32,7 @@ If you have other good projects, please refer to the above format to add to READ
### Environment Setup ### 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 ### Usage

View File

@ -4,7 +4,7 @@
下面以 [ADGEN](https://aclanthology.org/D19-1321.pdf) (广告生成) 数据集为例介绍代码的使用方法。 下面以 [ADGEN](https://aclanthology.org/D19-1321.pdf) (广告生成) 数据集为例介绍代码的使用方法。
## 软件依赖 ## 软件依赖
除 ChatGLM-6B 的依赖之外,还需要按照以下依赖 运行微调需要4.27.1版本的`transformers`。除 ChatGLM-6B 的依赖之外,还需要按照以下依赖
``` ```
pip install rouge_chinese nltk jieba datasets pip install rouge_chinese nltk jieba datasets
``` ```

View File

@ -1,5 +1,5 @@
protobuf>=3.19.5,<3.20.1 protobuf>=3.19.5,<3.20.1
transformers==4.26.1 transformers==4.27.1
icetk icetk
cpm_kernels cpm_kernels
torch>=1.10 torch>=1.10