add missing setting

This commit is contained in:
智丞
2022-06-30 14:25:44 +08:00
parent dadc07f492
commit 4a79f63a9d
2 changed files with 6 additions and 0 deletions

View File

@@ -10,4 +10,5 @@ from .multi_modal import OfaForImageCaptioning
from .nlp import (BertForMaskedLM, BertForSequenceClassification, SbertForNLI,
SbertForSentenceSimilarity, SbertForSentimentClassification,
SbertForTokenClassification, SbertForZeroShotClassification,
SpaceForDialogIntent, SpaceForDialogModeling,
StructBertForMaskedLM, VecoForMaskedLM)

View File

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