mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-24 03:59:23 +01:00
add missing setting
This commit is contained in:
@@ -10,4 +10,5 @@ from .multi_modal import OfaForImageCaptioning
|
||||
from .nlp import (BertForMaskedLM, BertForSequenceClassification, SbertForNLI,
|
||||
SbertForSentenceSimilarity, SbertForSentimentClassification,
|
||||
SbertForTokenClassification, SbertForZeroShotClassification,
|
||||
SpaceForDialogIntent, SpaceForDialogModeling,
|
||||
StructBertForMaskedLM, VecoForMaskedLM)
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user