diff --git a/modelscope/utils/constant.py b/modelscope/utils/constant.py index c26a9e24..3eb2890a 100644 --- a/modelscope/utils/constant.py +++ b/modelscope/utils/constant.py @@ -43,8 +43,7 @@ class Tasks(object): text_generation = 'text-generation' table_question_answering = 'table-question-answering' feature_extraction = 'feature-extraction' - sentence_similarity = 'sentence-similarity' - fill_mask = 'fill-mask ' + fill_mask = 'fill-mask' summarization = 'summarization' question_answering = 'question-answering'