Skip to content

Modify text summarization taskflow doc #5149

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
LazyFyh wants to merge 2 commits into from
Closed

Modify text summarization taskflow doc #5149

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a3b6580
Modify text summarization taskflow doc
LazyFyh Mar 8, 2023
cd5aebb
Add optional model list
LazyFyh Mar 8, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 14 additions & 1 deletion docs/model_zoo/taskflow.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1591,8 +1591,21 @@ from paddlenlp import Taskflow
```

#### 可配置参数说明
* `model`：可选模型，默认为`IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese`。
* `model`：可选模型，默认为"PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA"，支持的模型有["unimo-text-1.0-summary","IDEA-CCNL/Randeng-Pegasus-238M-Summary-Chinese", "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese", "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese-V1", "PaddlePaddle/Randeng-Pegasus-238M-Summary-Chinese-SSTIA", "PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA"]。
* `batch_size`：批处理大小，请结合机器情况进行调整，默认为1。
* `max_length`：生成摘要的最大文本长度，默认为128。
* `min_length`：生成摘要的最小文本长度，默认为0。
* `decode_strategy`：生成过程中的解码策略，默认为beam search。
* `temperature`：生成过程中softmax函数的超参数，默认为1.0。
* `top_k`：Top k采样中的保留结果数，默认为5。
* `top_p`：Top p采样中的累计概率，默认为1.0。
* `num_beams`：beam search中的beam数量，默认为4。
* `length_penalty`：生成摘要的文本长度惩罚，默认为0.0。
* `num_return_sequences`：单条输入返回的输出摘要个数，默认为1。
* `repetition_penalty`：生成摘要的文本重复惩罚，默认为1。
* `use_faster`：是否使用FasterGeneration高性能解码，默认为False。
* `use_fp16_decoding`：是否使用FasterGeneration中的FP16混合精度计算，默认为False。


</div></details>

Expand Down