Browse Source

update

master
suluyan 3 years ago
parent
commit
01ca8214fb
2 changed files with 4 additions and 6 deletions
  1. +1
    -1
      modelscope/pipelines/builder.py
  2. +3
    -5
      tests/pipelines/test_fill_mask.py

+ 1
- 1
modelscope/pipelines/builder.py View File

@@ -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'),
}


+ 3
- 5
tests/pipelines/test_fill_mask.py View File

@@ -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':


Loading…
Cancel
Save