suluyan
3 years ago
2 changed files with 3 additions and 1 deletions
Unified View
Diff Options
+ 1
- 0
modelscope/metainfo.py
View File
| @@ -45,6 +45,7 @@ class Pipelines(object): | |||||
| word_segmentation = 'word-segmentation' | word_segmentation = 'word-segmentation' | ||||
| text_generation = 'text-generation' | text_generation = 'text-generation' | ||||
| sentiment_analysis = 'sentiment-analysis' | sentiment_analysis = 'sentiment-analysis' | ||||
| fill_mask = 'fill-mask' | |||||
| # audio tasks | # audio tasks | ||||
| sambert_hifigan_16k_tts = 'sambert-hifigan-16k-tts' | sambert_hifigan_16k_tts = 'sambert-hifigan-16k-tts' | ||||
+ 2
- 1
modelscope/pipelines/nlp/fill_mask_pipeline.py
View File
| @@ -1,5 +1,6 @@ | |||||
| from typing import Dict, Optional, Union | from typing import Dict, Optional, Union | ||||
| from modelscope.metainfo import Pipelines | |||||
| from modelscope.models import Model | from modelscope.models import Model | ||||
| from modelscope.models.nlp.masked_language_model import \ | from modelscope.models.nlp.masked_language_model import \ | ||||
| AliceMindBaseForMaskedLM | AliceMindBaseForMaskedLM | ||||
| @@ -11,7 +12,7 @@ from ..builder import PIPELINES | |||||
| __all__ = ['FillMaskPipeline'] | __all__ = ['FillMaskPipeline'] | ||||
| @PIPELINES.register_module(Tasks.fill_mask, module_name=r'fill_mask') | |||||
| @PIPELINES.register_module(Tasks.fill_mask, module_name=Pipelines.fill_mask) | |||||
| class FillMaskPipeline(Pipeline): | class FillMaskPipeline(Pipeline): | ||||
| def __init__(self, | def __init__(self, | ||||