Skip to content

Commit 8e83083

Browse files
committed
Update README.md
1 parent 44c9b8b commit 8e83083

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

ptuning/README.md

+6
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,12 @@ model = model.eval()
178178
response, history = model.chat(tokenizer, "你好", history=[])
179179
```
180180

181+
**[23/04/19]** 你也可以直接运行支持加载 P-Tuning v2 checkpoint 的 [web demo](./web_demo.py)
182+
```shell
183+
bash web_demo.sh
184+
```
185+
可能需要修改 [web_demo.sh](./web_demo.sh) 的内容以符合你实际的 checkpoint 情况。
186+
181187
## 使用自己的数据集
182188
修改 `train.sh``evaluate.sh` 中的 `train_file``validation_file``test_file`为你自己的 JSON 格式数据集路径,并将 `prompt_column``response_column` 改为 JSON 文件中输入文本和输出文本对应的 KEY。可能还需要增大 `max_source_length``max_target_length` 来匹配你自己的数据集中的最大输入输出长度。
183189

0 commit comments

Comments
 (0)