Skip to content

Commit ca92e63

Browse files
committed
Update text_correction task name
1 parent af89ed9 commit ca92e63

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

docs/model_zoo/taskflow.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ from paddlenlp import Taskflow
632632
| `Taskflow("pos_tagging")` | `$HOME/.paddlenlp/taskflow/lac` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/lexical_analysis) |
633633
| `Taskflow("ner", mode="fast")` | `$HOME/.paddlenlp/taskflow/lac` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/lexical_analysis) |
634634
| `Taskflow("ner", mode="accurate")` | `$HOME/.paddlenlp/taskflow/wordtag` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/text_to_knowledge/ernie-ctm) |
635-
| `Taskflow("text_correction", model="csc-ernie-1.0")` | `$HOME/.paddlenlp/taskflow/text_correction/csc-ernie-1.0` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/text_correction/ernie-csc) |
635+
| `Taskflow("text_correction", model="ernie-csc")` | `$HOME/.paddlenlp/taskflow/text_correction/ernie-csc` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/text_correction/ernie-csc) |
636636
| `Taskflow("dependency_parsing", model="ddparser")` | `$HOME/.paddlenlp/taskflow/dependency_parsing/ddparser` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/dependency_parsing/ddparser) |
637637
| `Taskflow("dependency_parsing", model="ddparser-ernie-1.0")` | `$HOME/.paddlenlp/taskflow/dependency_parsing/ddparser-ernie-1.0` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/dependency_parsing/ddparser) |
638638
| `Taskflow("dependency_parsing", model="ddparser-ernie-gram-zh")` | `$HOME/.paddlenlp/taskflow/dependency_parsing/ddparser-ernie-gram-zh` | [示例](https://github.com/PaddlePaddle/PaddleNLP/tree/develop/examples/dependency_parsing/ddparser) |

paddlenlp/taskflow/taskflow.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -174,13 +174,13 @@
174174
},
175175
'text_correction': {
176176
"models": {
177-
"csc-ernie-1.0": {
177+
"ernie-csc": {
178178
"task_class": CSCTask,
179-
"task_flag": "text_correction-csc-ernie-1.0"
179+
"task_flag": "text_correction-ernie-csc"
180180
},
181181
},
182182
"default": {
183-
"model": "csc-ernie-1.0"
183+
"model": "ernie-csc"
184184
}
185185
},
186186
'text_similarity': {

paddlenlp/taskflow/text_correction.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
5959
"""
6060

61-
TASK_MODEL_MAP = {"csc-ernie-1.0": "ernie-1.0"}
61+
TASK_MODEL_MAP = {"ernie-csc": "ernie-1.0"}
6262

6363

6464
class CSCTask(Task):
@@ -75,13 +75,13 @@ class CSCTask(Task):
7575
"pinyin_vocab": "pinyin_vocab.txt"
7676
}
7777
resource_files_urls = {
78-
"csc-ernie-1.0": {
78+
"ernie-csc": {
7979
"model_state": [
80-
"https://bj.bcebos.com/paddlenlp/taskflow/text_correction/csc-ernie-1.0/model_state.pdparams",
80+
"https://bj.bcebos.com/paddlenlp/taskflow/text_correction/ernie-csc/model_state.pdparams",
8181
"cdc53e7e3985ffc78fedcdf8e6dca6d2"
8282
],
8383
"pinyin_vocab": [
84-
"https://bj.bcebos.com/paddlenlp/taskflow/text_correction/csc-ernie-1.0/pinyin_vocab.txt",
84+
"https://bj.bcebos.com/paddlenlp/taskflow/text_correction/ernie-csc/pinyin_vocab.txt",
8585
"5599a8116b6016af573d08f8e686b4b2"
8686
],
8787
}

0 commit comments

Comments
 (0)