Browse Source

add missing setting

master
智丞 3 years ago
parent
commit
4a79f63a9d
2 changed files with 6 additions and 0 deletions
  1. +1
    -0
      modelscope/models/__init__.py
  2. +5
    -0
      modelscope/pipelines/builder.py

+ 1
- 0
modelscope/models/__init__.py 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)

+ 5
- 0
modelscope/pipelines/builder.py 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:


Loading…
Cancel
Save