From 01ca8214fbc3119026d200ea4270dea8335b1789 Mon Sep 17 00:00:00 2001 From: suluyan Date: Thu, 23 Jun 2022 18:15:40 +0800 Subject: [PATCH] update --- modelscope/pipelines/builder.py | 2 +- tests/pipelines/test_fill_mask.py | 8 +++----- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/modelscope/pipelines/builder.py b/modelscope/pipelines/builder.py index 2e8d9379..d3be06bc 100644 --- a/modelscope/pipelines/builder.py +++ b/modelscope/pipelines/builder.py @@ -34,7 +34,7 @@ DEFAULT_MODEL_FOR_PIPELINE = { 'damo/cv_unet_person-image-cartoon_compound-models'), Tasks.ocr_detection: (Pipelines.ocr_detection, 'damo/cv_resnet18_ocr-detection-line-level_damo'), - Tasks.fill_mask: (Pipelines.fill_mask, 'damo/nlp_veco_fill-mask_large'), + Tasks.fill_mask: (Pipelines.fill_mask, 'damo/nlp_veco_fill-mask-large'), Tasks.action_recognition: (Pipelines.action_recognition, 'damo/cv_TAdaConv_action-recognition'), } diff --git a/tests/pipelines/test_fill_mask.py b/tests/pipelines/test_fill_mask.py index 649012e0..87f71b31 100644 --- a/tests/pipelines/test_fill_mask.py +++ b/tests/pipelines/test_fill_mask.py @@ -1,6 +1,4 @@ # Copyright (c) Alibaba, Inc. and its affiliates. -import os -import shutil import unittest from modelscope.hub.snapshot_download import snapshot_download @@ -14,10 +12,10 @@ from modelscope.utils.test_utils import test_level class FillMaskTest(unittest.TestCase): model_id_sbert = { - 'zh': 'damo/nlp_structbert_fill-mask-chinese_large', - 'en': 'damo/nlp_structbert_fill-mask-english_large' + 'zh': 'damo/nlp_structbert_fill-mask_chinese-large', + 'en': 'damo/nlp_structbert_fill-mask_english-large' } - model_id_veco = 'damo/nlp_veco_fill-mask_large' + model_id_veco = 'damo/nlp_veco_fill-mask-large' ori_texts = { 'zh':