From 8e83083d7fe0dd1a6b7f828b1332bab0d9412ae8 Mon Sep 17 00:00:00 2001 From: duzx16 Date: Wed, 19 Apr 2023 23:00:29 +0800 Subject: [PATCH] Update README.md --- ptuning/README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/ptuning/README.md b/ptuning/README.md index e3339ce..2c828b3 100644 --- a/ptuning/README.md +++ b/ptuning/README.md @@ -178,6 +178,12 @@ model = model.eval() response, history = model.chat(tokenizer, "你好", history=[]) ``` +**[23/04/19]** 你也可以直接运行支持加载 P-Tuning v2 checkpoint 的 [web demo](./web_demo.py) +```shell +bash web_demo.sh +``` +可能需要修改 [web_demo.sh](./web_demo.sh) 的内容以符合你实际的 checkpoint 情况。 + ## 使用自己的数据集 修改 `train.sh` 和 `evaluate.sh` 中的 `train_file`、`validation_file`和`test_file`为你自己的 JSON 格式数据集路径，并将 `prompt_column` 和 `response_column` 改为 JSON 文件中输入文本和输出文本对应的 KEY。可能还需要增大 `max_source_length` 和 `max_target_length` 来匹配你自己的数据集中的最大输入输出长度。