mirror of https://github.com/THUDM/ChatGLM-6B
Update requirements.txt
parent
24e24d5d6c
commit
c206e7d9ad
|
@ -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` 即可。
|
||||
|
||||
### 代码调用
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
```
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue