Skip to content

Commit a345ca7

Browse files
author
gongenlei
authored
add models for summ (#4933)
1 parent b1428e6 commit a345ca7

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

paddlenlp/taskflow/taskflow.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,8 +255,23 @@
255255
"task_flag": "text_summarization-IDEA-CCNL/Randeng-Pegasus523M-Summary-Chinese",
256256
"task_priority_path": "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese",
257257
},
258+
"IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese-V1": {
259+
"task_class": TextSummarizationTask,
260+
"task_flag": "text_summarization-IDEA-CCNL/Randeng-Pegasus523M-Summary-Chinese-V1",
261+
"task_priority_path": "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese-V1",
262+
},
263+
"PaddlePaddle/Randeng-Pegasus-238M-Summary-Chinese-SSTIA": {
264+
"task_class": TextSummarizationTask,
265+
"task_flag": "text_summarization-PaddlePaddle/Randeng-Pegasus-238M-Summary-Chinese-SSTIA",
266+
"task_priority_path": "PaddlePaddle/Randeng-Pegasus-238M-Summary-Chinese-SSTIA",
267+
},
268+
"PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA": {
269+
"task_class": TextSummarizationTask,
270+
"task_flag": "text_summarization-PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA",
271+
"task_priority_path": "PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA",
272+
},
258273
},
259-
"default": {"model": "IDEA-CCNL/Randeng-Pegasus-523M-Summary-Chinese"},
274+
"default": {"model": "PaddlePaddle/Randeng-Pegasus-523M-Summary-Chinese-SSTIA"},
260275
},
261276
"word_segmentation": {
262277
"modes": {

0 commit comments

Comments
 (0)