diff --git a/modelscope/models/__init__.py b/modelscope/models/__init__.py index 0a69e6c8..a4b9640d 100644 --- a/modelscope/models/__init__.py +++ b/modelscope/models/__init__.py @@ -10,4 +10,5 @@ from .multi_modal import OfaForImageCaptioning from .nlp import (BertForMaskedLM, BertForSequenceClassification, SbertForNLI, SbertForSentenceSimilarity, SbertForSentimentClassification, SbertForTokenClassification, SbertForZeroShotClassification, + SpaceForDialogIntent, SpaceForDialogModeling, StructBertForMaskedLM, VecoForMaskedLM) diff --git a/modelscope/pipelines/builder.py b/modelscope/pipelines/builder.py index 5f564d0b..36f87269 100644 --- a/modelscope/pipelines/builder.py +++ b/modelscope/pipelines/builder.py @@ -36,6 +36,11 @@ DEFAULT_MODEL_FOR_PIPELINE = { Tasks.zero_shot_classification: (Pipelines.zero_shot_classification, 'damo/nlp_structbert_zero-shot-classification_chinese-base'), + Tasks.dialog_intent_prediction: + (Pipelines.dialog_intent_prediction, + 'damo/nlp_space_dialog-intent-prediction'), + Tasks.dialog_modeling: (Pipelines.dialog_modeling, + 'damo/nlp_space_dialog-modeling'), Tasks.image_captioning: (Pipelines.image_caption, 'damo/ofa_image-caption_coco_large_en'), Tasks.image_generation: