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
|
### 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
|
||||||
|
|
||||||
|
|
|
@ -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
|
||||||
```
|
```
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue