We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 44c9b8b commit 8e83083Copy full SHA for 8e83083
ptuning/README.md
@@ -178,6 +178,12 @@ model = model.eval()
178
response, history = model.chat(tokenizer, "你好", history=[])
179
```
180
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
+
187
## 使用自己的数据集
188
修改 `train.sh` 和 `evaluate.sh` 中的 `train_file`、`validation_file`和`test_file`为你自己的 JSON 格式数据集路径,并将 `prompt_column` 和 `response_column` 改为 JSON 文件中输入文本和输出文本对应的 KEY。可能还需要增大 `max_source_length` 和 `max_target_length` 来匹配你自己的数据集中的最大输入输出长度。
189
0 commit comments