zhangzhicheng.zzc yingda.chen 3 years ago
parent
commit
016d22d8fb
66 changed files with 249 additions and 262 deletions
  1. +4
    -3
      modelscope/hub/api.py
  2. +1
    -1
      modelscope/hub/file_download.py
  3. +1
    -1
      modelscope/hub/repository.py
  4. +1
    -1
      modelscope/hub/snapshot_download.py
  5. +4
    -4
      modelscope/metrics/builder.py
  6. +4
    -3
      modelscope/metrics/sequence_classification_metric.py
  7. +2
    -2
      modelscope/metrics/text_generation_metric.py
  8. +1
    -1
      modelscope/models/audio/ans/frcrn.py
  9. +2
    -2
      modelscope/models/multi_modal/image_captioning_model.py
  10. +4
    -4
      modelscope/models/multi_modal/mplug_for_visual_question_answering.py
  11. +2
    -2
      modelscope/models/multi_modal/ofa_for_image_captioning_model.py
  12. +1
    -1
      modelscope/models/nlp/backbones/space/model/intent_unified_transformer.py
  13. +1
    -1
      modelscope/models/nlp/backbones/space/model/model_base.py
  14. +1
    -1
      modelscope/models/nlp/backbones/structbert/adv_utils.py
  15. +1
    -1
      modelscope/models/nlp/backbones/structbert/configuration_sbert.py
  16. +5
    -5
      modelscope/models/nlp/backbones/structbert/modeling_sbert.py
  17. +4
    -4
      modelscope/models/nlp/bert_for_sequence_classification.py
  18. +4
    -4
      modelscope/models/nlp/csanmt_for_translation.py
  19. +5
    -5
      modelscope/models/nlp/heads/sequence_classification_head.py
  20. +4
    -4
      modelscope/models/nlp/masked_language.py
  21. +4
    -4
      modelscope/models/nlp/nncrf_for_named_entity_recognition.py
  22. +4
    -4
      modelscope/models/nlp/palm_for_text_generation.py
  23. +3
    -3
      modelscope/models/nlp/sbert_for_nli.py
  24. +3
    -3
      modelscope/models/nlp/sbert_for_sentence_similarity.py
  25. +3
    -3
      modelscope/models/nlp/sbert_for_sentiment_classification.py
  26. +1
    -4
      modelscope/models/nlp/sbert_for_sequence_classification.py
  27. +4
    -4
      modelscope/models/nlp/sbert_for_token_classification.py
  28. +3
    -3
      modelscope/models/nlp/sbert_for_zero_shot_classification.py
  29. +4
    -4
      modelscope/models/nlp/sequence_classification.py
  30. +7
    -7
      modelscope/models/nlp/space_for_dialog_intent_prediction.py
  31. +8
    -8
      modelscope/models/nlp/space_for_dialog_modeling.py
  32. +4
    -4
      modelscope/models/nlp/space_for_dialog_state_tracking.py
  33. +9
    -9
      modelscope/models/nlp/task_model.py
  34. +2
    -2
      modelscope/pipelines/audio/ans_pipeline.py
  35. +2
    -2
      modelscope/pipelines/audio/linear_aec_pipeline.py
  36. +2
    -3
      modelscope/pipelines/cv/action_recognition_pipeline.py
  37. +2
    -3
      modelscope/pipelines/cv/animal_recog_pipeline.py
  38. +2
    -3
      modelscope/pipelines/cv/cmdssl_video_embedding_pipleline.py
  39. +2
    -3
      modelscope/pipelines/cv/face_image_generation_pipeline.py
  40. +2
    -3
      modelscope/pipelines/cv/image_cartoon_pipeline.py
  41. +2
    -3
      modelscope/pipelines/cv/image_colorization_pipeline.py
  42. +2
    -3
      modelscope/pipelines/cv/image_matting_pipeline.py
  43. +2
    -3
      modelscope/pipelines/cv/image_super_resolution_pipeline.py
  44. +2
    -3
      modelscope/pipelines/cv/ocr_detection_pipeline.py
  45. +4
    -4
      modelscope/pipelines/multi_modal/image_captioning_pipeline.py
  46. +2
    -3
      modelscope/pipelines/multi_modal/multi_modal_embedding_pipeline.py
  47. +2
    -3
      modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py
  48. +7
    -7
      modelscope/pipelines/multi_modal/visual_question_answering_pipeline.py
  49. +8
    -8
      modelscope/pipelines/nlp/dialog_intent_prediction_pipeline.py
  50. +8
    -8
      modelscope/pipelines/nlp/dialog_modeling_pipeline.py
  51. +7
    -7
      modelscope/pipelines/nlp/dialog_state_tracking_pipeline.py
  52. +8
    -8
      modelscope/pipelines/nlp/fill_mask_pipeline.py
  53. +8
    -8
      modelscope/pipelines/nlp/named_entity_recognition_pipeline.py
  54. +7
    -7
      modelscope/pipelines/nlp/nli_pipeline.py
  55. +7
    -7
      modelscope/pipelines/nlp/sentence_similarity_pipeline.py
  56. +7
    -7
      modelscope/pipelines/nlp/sentiment_classification_pipeline.py
  57. +3
    -3
      modelscope/pipelines/nlp/sequence_classification_pipeline.py
  58. +7
    -7
      modelscope/pipelines/nlp/text_generation_pipeline.py
  59. +7
    -7
      modelscope/pipelines/nlp/translation_pipeline.py
  60. +7
    -7
      modelscope/pipelines/nlp/word_segmentation_pipeline.py
  61. +7
    -7
      modelscope/pipelines/nlp/zero_shot_classification_pipeline.py
  62. +5
    -5
      modelscope/preprocessors/nlp.py
  63. +2
    -3
      modelscope/trainers/trainer.py
  64. +3
    -3
      modelscope/trainers/utils/inference.py
  65. +1
    -1
      modelscope/utils/tensor_utils.py
  66. +1
    -1
      modelscope/utils/utils.py

+ 4
- 3
modelscope/hub/api.py View File

@@ -9,10 +9,11 @@ from typing import List, Optional, Tuple, Union


import requests import requests


from modelscope.msdatasets.config import (DOWNLOADED_DATASETS_PATH,
HUB_DATASET_ENDPOINT)
from modelscope.utils.constant import (DEFAULT_DATASET_REVISION,
DEFAULT_MODEL_REVISION, DownloadMode)
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..msdatasets.config import DOWNLOADED_DATASETS_PATH, HUB_DATASET_ENDPOINT
from ..utils.constant import (DEFAULT_DATASET_REVISION, DEFAULT_MODEL_REVISION,
DownloadMode)
from .errors import (InvalidParameter, NotExistError, RequestError, from .errors import (InvalidParameter, NotExistError, RequestError,
datahub_raise_on_error, handle_http_response, is_ok, datahub_raise_on_error, handle_http_response, is_ok,
raise_on_error) raise_on_error)


+ 1
- 1
modelscope/hub/file_download.py View File

@@ -19,8 +19,8 @@ from requests.exceptions import HTTPError
from tqdm import tqdm from tqdm import tqdm


from modelscope import __version__ from modelscope import __version__
from modelscope.utils.constant import DEFAULT_MODEL_REVISION
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..utils.constant import DEFAULT_MODEL_REVISION
from .api import HubApi, ModelScopeConfig from .api import HubApi, ModelScopeConfig
from .constants import (DEFAULT_MODELSCOPE_GROUP, LOGGER_NAME, from .constants import (DEFAULT_MODELSCOPE_GROUP, LOGGER_NAME,
MODEL_ID_SEPARATOR) MODEL_ID_SEPARATOR)


+ 1
- 1
modelscope/hub/repository.py View File

@@ -2,8 +2,8 @@ import os
from typing import Optional from typing import Optional


from modelscope.hub.errors import GitError, InvalidParameter from modelscope.hub.errors import GitError, InvalidParameter
from modelscope.utils.constant import DEFAULT_MODEL_REVISION
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..utils.constant import DEFAULT_MODEL_REVISION
from .api import ModelScopeConfig from .api import ModelScopeConfig
from .git import GitCommandWrapper from .git import GitCommandWrapper
from .utils.utils import get_endpoint from .utils.utils import get_endpoint


+ 1
- 1
modelscope/hub/snapshot_download.py View File

@@ -3,8 +3,8 @@ import tempfile
from pathlib import Path from pathlib import Path
from typing import Dict, Optional, Union from typing import Dict, Optional, Union


from modelscope.utils.constant import DEFAULT_MODEL_REVISION
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..utils.constant import DEFAULT_MODEL_REVISION
from .api import HubApi, ModelScopeConfig from .api import HubApi, ModelScopeConfig
from .errors import NotExistError from .errors import NotExistError
from .file_download import (get_file_download_url, http_get_file, from .file_download import (get_file_download_url, http_get_file,


+ 4
- 4
modelscope/metrics/builder.py View File

@@ -1,9 +1,9 @@
# Copyright (c) Alibaba, Inc. and its affiliates. # Copyright (c) Alibaba, Inc. and its affiliates.


from ..metainfo import Metrics
from ..utils.config import ConfigDict
from ..utils.constant import Tasks
from ..utils.registry import Registry, build_from_cfg, default_group
from modelscope.metainfo import Metrics
from modelscope.utils.config import ConfigDict
from modelscope.utils.constant import Tasks
from modelscope.utils.registry import Registry, build_from_cfg, default_group


METRICS = Registry('metrics') METRICS = Registry('metrics')




+ 4
- 3
modelscope/metrics/sequence_classification_metric.py View File

@@ -2,10 +2,11 @@ from typing import Dict, List, Union


import numpy as np import numpy as np


from modelscope.metainfo import Metrics
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ..metainfo import Metrics
from ..utils.registry import default_group
from ..utils.tensor_utils import torch_nested_detach, torch_nested_numpify
from modelscope.utils.registry import default_group
from modelscope.utils.tensor_utils import (torch_nested_detach,
torch_nested_numpify)
from .base import Metric from .base import Metric
from .builder import METRICS, MetricKeys from .builder import METRICS, MetricKeys




+ 2
- 2
modelscope/metrics/text_generation_metric.py View File

@@ -1,7 +1,7 @@
from typing import Dict from typing import Dict


from ..metainfo import Metrics
from ..utils.registry import default_group
from modelscope.metainfo import Metrics
from modelscope.utils.registry import default_group
from .base import Metric from .base import Metric
from .builder import METRICS, MetricKeys from .builder import METRICS, MetricKeys




+ 1
- 1
modelscope/models/audio/ans/frcrn.py View File

@@ -6,9 +6,9 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F


from modelscope.metainfo import Models from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS from modelscope.models.builder import MODELS
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from ...base import Model, Tensor
from .conv_stft import ConviSTFT, ConvSTFT from .conv_stft import ConviSTFT, ConvSTFT
from .unet import UNet from .unet import UNet




+ 2
- 2
modelscope/models/multi_modal/image_captioning_model.py View File

@@ -5,9 +5,9 @@ import torch.cuda
from PIL import Image from PIL import Image


from modelscope.metainfo import Models from modelscope.metainfo import Models
from modelscope.models.base import Model
from modelscope.models.builder import MODELS
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from ..base import Model
from ..builder import MODELS


__all__ = ['OfaForImageCaptioning'] __all__ = ['OfaForImageCaptioning']




+ 4
- 4
modelscope/models/multi_modal/mplug_for_visual_question_answering.py View File

@@ -1,9 +1,9 @@
from typing import Dict from typing import Dict


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['MPlugForVisualQuestionAnswering'] __all__ = ['MPlugForVisualQuestionAnswering']




+ 2
- 2
modelscope/models/multi_modal/ofa_for_image_captioning_model.py View File

@@ -3,10 +3,10 @@ from typing import Any, Dict
import torch.cuda import torch.cuda


from modelscope.metainfo import Models from modelscope.metainfo import Models
from modelscope.models.base import Model
from modelscope.models.builder import MODELS
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from ..base import Model
from ..builder import MODELS
from .ofa import OFAModel, OFATokenizer from .ofa import OFAModel, OFATokenizer
from .ofa.generate import sequence_generator as sg from .ofa.generate import sequence_generator as sg
from .ofa.generate.utils import move_to_device from .ofa.generate.utils import move_to_device


+ 1
- 1
modelscope/models/nlp/backbones/space/model/intent_unified_transformer.py View File

@@ -4,7 +4,7 @@ import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F


from ......utils.nlp.space.criterions import compute_kl_loss
from modelscope.utils.nlp.space.criterions import compute_kl_loss
from .unified_transformer import UnifiedTransformer from .unified_transformer import UnifiedTransformer






+ 1
- 1
modelscope/models/nlp/backbones/space/model/model_base.py View File

@@ -4,7 +4,7 @@ import os


import torch.nn as nn import torch.nn as nn


from ......utils.constant import ModelFile
from modelscope.utils.constant import ModelFile




class SpaceModelBase(nn.Module): class SpaceModelBase(nn.Module):


+ 1
- 1
modelscope/models/nlp/backbones/structbert/adv_utils.py View File

@@ -16,7 +16,7 @@
import torch import torch
from torch import nn from torch import nn


from .....utils.logger import get_logger
from modelscope.utils.logger import get_logger


logger = get_logger(__name__) logger = get_logger(__name__)




+ 1
- 1
modelscope/models/nlp/backbones/structbert/configuration_sbert.py View File

@@ -17,7 +17,7 @@
""" SBERT model configuration, mainly copied from :class:`~transformers.BertConfig` """ """ SBERT model configuration, mainly copied from :class:`~transformers.BertConfig` """
from transformers import PretrainedConfig from transformers import PretrainedConfig


from .....utils import logger as logging
from modelscope.utils import logger as logging


logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)




+ 5
- 5
modelscope/models/nlp/backbones/structbert/modeling_sbert.py View File

@@ -15,11 +15,11 @@ from transformers.modeling_utils import (apply_chunking_to_forward,
find_pruneable_heads_and_indices, find_pruneable_heads_and_indices,
prune_linear_layer) prune_linear_layer)


from .....metainfo import Models
from .....utils.constant import Fields
from .....utils.logger import get_logger
from ....base import TorchModel
from ....builder import BACKBONES
from modelscope.metainfo import Models
from modelscope.models.base import TorchModel
from modelscope.models.builder import BACKBONES
from modelscope.utils.constant import Fields
from modelscope.utils.logger import get_logger
from .configuration_sbert import SbertConfig from .configuration_sbert import SbertConfig


logger = get_logger(__name__) logger = get_logger(__name__)


+ 4
- 4
modelscope/models/nlp/bert_for_sequence_classification.py View File

@@ -4,10 +4,10 @@ from typing import Any, Dict
import json import json
import numpy as np import numpy as np


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Model
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['BertForSequenceClassification'] __all__ = ['BertForSequenceClassification']




+ 4
- 4
modelscope/models/nlp/csanmt_for_translation.py View File

@@ -7,10 +7,10 @@ import json
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['CsanmtForTranslation'] __all__ = ['CsanmtForTranslation']




+ 5
- 5
modelscope/models/nlp/heads/sequence_classification_head.py View File

@@ -5,11 +5,11 @@ import torch
import torch.nn.functional as F import torch.nn.functional as F
from torch import nn from torch import nn


from ....metainfo import Heads
from ....outputs import OutputKeys
from ....utils.constant import Tasks
from ...base import TorchHead
from ...builder import HEADS
from modelscope.metainfo import Heads
from modelscope.models.base import TorchHead
from modelscope.models.builder import HEADS
from modelscope.outputs import OutputKeys
from modelscope.utils.constant import Tasks




@HEADS.register_module( @HEADS.register_module(


+ 4
- 4
modelscope/models/nlp/masked_language.py View File

@@ -2,10 +2,10 @@ from typing import Any, Dict, Optional, Union


import numpy as np import numpy as np


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['BertForMaskedLM', 'StructBertForMaskedLM', 'VecoForMaskedLM'] __all__ = ['BertForMaskedLM', 'StructBertForMaskedLM', 'VecoForMaskedLM']




+ 4
- 4
modelscope/models/nlp/nncrf_for_named_entity_recognition.py View File

@@ -8,10 +8,10 @@ import torch.nn as nn
from torch.autograd import Variable from torch.autograd import Variable
from transformers import AutoConfig, AutoModel from transformers import AutoConfig, AutoModel


from ...metainfo import Models
from ...utils.constant import ModelFile, Tasks
from ..base import Model
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model
from modelscope.models.builder import MODELS
from modelscope.utils.constant import ModelFile, Tasks


__all__ = ['TransformerCRFForNamedEntityRecognition'] __all__ = ['TransformerCRFForNamedEntityRecognition']




+ 4
- 4
modelscope/models/nlp/palm_for_text_generation.py View File

@@ -1,9 +1,9 @@
from typing import Dict from typing import Dict


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Tensor, TorchModel
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Tensor, TorchModel
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['PalmForTextGeneration'] __all__ = ['PalmForTextGeneration']




+ 3
- 3
modelscope/models/nlp/sbert_for_nli.py View File

@@ -1,6 +1,6 @@
from ...metainfo import Models
from ...utils.constant import Tasks
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks
from .sbert_for_sequence_classification import \ from .sbert_for_sequence_classification import \
SbertForSequenceClassificationBase SbertForSequenceClassificationBase




+ 3
- 3
modelscope/models/nlp/sbert_for_sentence_similarity.py View File

@@ -1,6 +1,6 @@
from ...metainfo import Models
from ...utils.constant import Tasks
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks
from .sbert_for_sequence_classification import \ from .sbert_for_sequence_classification import \
SbertForSequenceClassificationBase SbertForSequenceClassificationBase




+ 3
- 3
modelscope/models/nlp/sbert_for_sentiment_classification.py View File

@@ -1,6 +1,6 @@
from ...metainfo import Models
from ...utils.constant import Tasks
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks
from .sbert_for_sequence_classification import \ from .sbert_for_sequence_classification import \
SbertForSequenceClassificationBase SbertForSequenceClassificationBase




+ 1
- 4
modelscope/models/nlp/sbert_for_sequence_classification.py View File

@@ -7,10 +7,7 @@ import torch
from sofa.models.sbert.modeling_sbert import SbertModel, SbertPreTrainedModel from sofa.models.sbert.modeling_sbert import SbertModel, SbertPreTrainedModel
from torch import nn from torch import nn


from modelscope.metainfo import Models
from modelscope.utils.constant import Tasks
from ..base import Model
from ..builder import MODELS
from modelscope.models.base import Model




class SbertTextClassfier(SbertPreTrainedModel): class SbertTextClassfier(SbertPreTrainedModel):


+ 4
- 4
modelscope/models/nlp/sbert_for_token_classification.py View File

@@ -3,10 +3,10 @@ from typing import Any, Dict, Union
import numpy as np import numpy as np
import torch import torch


from ...metainfo import Models
from ...utils.constant import Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks


__all__ = ['SbertForTokenClassification'] __all__ = ['SbertForTokenClassification']




+ 3
- 3
modelscope/models/nlp/sbert_for_zero_shot_classification.py View File

@@ -2,10 +2,10 @@ from typing import Any, Dict


import numpy as np import numpy as np


from modelscope.metainfo import Models
from modelscope.models.base import Model
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from ...metainfo import Models
from ..base import Model
from ..builder import MODELS


__all__ = ['SbertForZeroShotClassification'] __all__ = ['SbertForZeroShotClassification']




+ 4
- 4
modelscope/models/nlp/sequence_classification.py View File

@@ -4,10 +4,10 @@ from typing import Any, Dict
import json import json
import numpy as np import numpy as np


from ...metainfo import TaskModels
from ...outputs import OutputKeys
from ...utils.constant import Tasks
from ..builder import MODELS
from modelscope.metainfo import TaskModels
from modelscope.models.builder import MODELS
from modelscope.outputs import OutputKeys
from modelscope.utils.constant import Tasks
from .task_model import SingleBackboneTaskModelBase from .task_model import SingleBackboneTaskModelBase


__all__ = ['SequenceClassificationModel'] __all__ = ['SequenceClassificationModel']


+ 7
- 7
modelscope/models/nlp/space_for_dialog_intent_prediction.py View File

@@ -4,13 +4,13 @@ import os
from typing import Any, Dict from typing import Any, Dict


from modelscope.metainfo import Models from modelscope.metainfo import Models
from modelscope.models.nlp.backbones.space import (SpaceGenerator,
SpaceModelBase)
from ...preprocessors.space.fields.intent_field import IntentBPETextField
from ...utils.config import Config
from ...utils.constant import ModelFile, Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.preprocessors.space.fields.intent_field import \
IntentBPETextField
from modelscope.utils.config import Config
from modelscope.utils.constant import ModelFile, Tasks
from .backbones import SpaceGenerator, SpaceModelBase


__all__ = ['SpaceForDialogIntent'] __all__ = ['SpaceForDialogIntent']




+ 8
- 8
modelscope/models/nlp/space_for_dialog_modeling.py View File

@@ -3,14 +3,14 @@
import os import os
from typing import Any, Dict, Optional from typing import Any, Dict, Optional


from modelscope.models.nlp.backbones.space import (SpaceGenerator,
SpaceModelBase)
from ...metainfo import Models
from ...preprocessors.space.fields.gen_field import MultiWOZBPETextField
from ...utils.config import Config
from ...utils.constant import ModelFile, Tasks
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.preprocessors.space.fields.gen_field import \
MultiWOZBPETextField
from modelscope.utils.config import Config
from modelscope.utils.constant import ModelFile, Tasks
from .backbones import SpaceGenerator, SpaceModelBase


__all__ = ['SpaceForDialogModeling'] __all__ = ['SpaceForDialogModeling']




+ 4
- 4
modelscope/models/nlp/space_for_dialog_state_tracking.py View File

@@ -1,11 +1,11 @@
import os import os
from typing import Any, Dict from typing import Any, Dict


from modelscope.metainfo import Models
from modelscope.models.base import Model, Tensor
from modelscope.models.builder import MODELS
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from ...metainfo import Models
from ...utils.nlp.space.utils_dst import batch_to_device
from ..base import Model, Tensor
from ..builder import MODELS
from modelscope.utils.nlp.space.utils_dst import batch_to_device


__all__ = ['SpaceForDialogStateTracking'] __all__ = ['SpaceForDialogStateTracking']




+ 9
- 9
modelscope/models/nlp/task_model.py View File

@@ -7,12 +7,12 @@ from typing import Any, Dict
import torch import torch
from torch import nn from torch import nn


from ...utils.config import ConfigDict
from ...utils.constant import Fields, Tasks
from ...utils.logger import get_logger
from ...utils.utils import if_func_recieve_dict_inputs
from ..base import TorchModel
from ..builder import build_backbone, build_head
from modelscope.models.base import TorchModel
from modelscope.models.builder import build_backbone, build_head
from modelscope.utils.config import ConfigDict
from modelscope.utils.constant import Fields, Tasks
from modelscope.utils.logger import get_logger
from modelscope.utils.utils import if_func_receive_dict_inputs


logger = get_logger(__name__) logger = get_logger(__name__)


@@ -424,7 +424,7 @@ class SingleBackboneTaskModelBase(BaseTaskModel):


def forward(self, input: Dict[str, Any]) -> Dict[str, Any]: def forward(self, input: Dict[str, Any]) -> Dict[str, Any]:
"""default forward method is the backbone-only forward""" """default forward method is the backbone-only forward"""
if if_func_recieve_dict_inputs(self.backbone.forward, input):
if if_func_receive_dict_inputs(self.backbone.forward, input):
outputs = self.backbone.forward(input) outputs = self.backbone.forward(input)
else: else:
outputs = self.backbone.forward(**input) outputs = self.backbone.forward(**input)
@@ -472,13 +472,13 @@ class EncoderDecoderTaskModelBase(BaseTaskModel):
return getattr(self, self._decoder_prefix) return getattr(self, self._decoder_prefix)


def forward(self, input: Dict[str, Any]) -> Dict[str, Any]: def forward(self, input: Dict[str, Any]) -> Dict[str, Any]:
if if_func_recieve_dict_inputs(self.encoder_.forward, input):
if if_func_receive_dict_inputs(self.encoder_.forward, input):
encoder_outputs = self.encoder_.forward(input) encoder_outputs = self.encoder_.forward(input)
else: else:
encoder_outputs = self.encoder_.forward(**input) encoder_outputs = self.encoder_.forward(**input)
decoder_inputs = self.project_decoder_inputs_and_mediate( decoder_inputs = self.project_decoder_inputs_and_mediate(
input, encoder_outputs) input, encoder_outputs)
if if_func_recieve_dict_inputs(self.decoder_.forward, input):
if if_func_receive_dict_inputs(self.decoder_.forward, input):
outputs = self.decoder_.forward(decoder_inputs) outputs = self.decoder_.forward(decoder_inputs)
else: else:
outputs = self.decoder_.forward(**decoder_inputs) outputs = self.decoder_.forward(**decoder_inputs)


+ 2
- 2
modelscope/pipelines/audio/ans_pipeline.py View File

@@ -8,10 +8,10 @@ import torch


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.torch_utils import create_device from modelscope.utils.torch_utils import create_device
from ..base import Input, Pipeline
from ..builder import PIPELINES




def audio_norm(x): def audio_norm(x):


+ 2
- 2
modelscope/pipelines/audio/linear_aec_pipeline.py View File

@@ -10,11 +10,11 @@ import yaml
from modelscope.fileio import File from modelscope.fileio import File
from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors.audio import LinearAECAndFbank from modelscope.preprocessors.audio import LinearAECAndFbank
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/action_recognition_pipeline.py View File

@@ -7,13 +7,12 @@ import torch
from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.models.cv.action_recognition.models import BaseVideoModel from modelscope.models.cv.action_recognition.models import BaseVideoModel
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors.video import ReadVideoData from modelscope.preprocessors.video import ReadVideoData
from modelscope.utils.config import Config from modelscope.utils.config import Config
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/animal_recog_pipeline.py View File

@@ -11,12 +11,11 @@ from modelscope.hub.snapshot_download import snapshot_download
from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.models.cv.animal_recognition import resnet from modelscope.models.cv.animal_recognition import resnet
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/cmdssl_video_embedding_pipleline.py View File

@@ -10,12 +10,11 @@ from modelscope.metainfo import Pipelines
from modelscope.models.cv.cmdssl_video_embedding.resnet2p1d import \ from modelscope.models.cv.cmdssl_video_embedding.resnet2p1d import \
resnet26_2p1d resnet26_2p1d
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.utils.config import Config from modelscope.utils.config import Config
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/face_image_generation_pipeline.py View File

@@ -9,12 +9,11 @@ import torch
from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.models.cv.face_generation import stylegan2 from modelscope.models.cv.face_generation import stylegan2
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/image_cartoon_pipeline.py View File

@@ -12,12 +12,11 @@ from modelscope.models.cv.cartoon.mtcnn_pytorch.src.align_trans import (
get_reference_facial_points, warp_and_crop_face) get_reference_facial_points, warp_and_crop_face)
from modelscope.models.cv.cartoon.utils import get_f5p, padTo16x, resize_size from modelscope.models.cv.cartoon.utils import get_f5p, padTo16x, resize_size
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


if tf.__version__ >= '2.0': if tf.__version__ >= '2.0':
tf = tf.compat.v1 tf = tf.compat.v1


+ 2
- 3
modelscope/pipelines/cv/image_colorization_pipeline.py View File

@@ -10,12 +10,11 @@ from modelscope.metainfo import Pipelines
from modelscope.models.cv.image_colorization import unet from modelscope.models.cv.image_colorization import unet
from modelscope.models.cv.image_colorization.utils import NormType from modelscope.models.cv.image_colorization.utils import NormType
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/image_matting_pipeline.py View File

@@ -7,12 +7,11 @@ import PIL


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/image_super_resolution_pipeline.py View File

@@ -8,12 +8,11 @@ import torch
from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.models.cv.super_resolution import rrdbnet_arch from modelscope.models.cv.super_resolution import rrdbnet_arch
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/cv/ocr_detection_pipeline.py View File

@@ -8,12 +8,11 @@ import tensorflow as tf


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import load_image from modelscope.preprocessors import load_image
from modelscope.utils.constant import ModelFile, Tasks from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Pipeline
from ..builder import PIPELINES
from .ocr_utils import model_resnet_mutex_v4_linewithchar, ops, utils from .ocr_utils import model_resnet_mutex_v4_linewithchar, ops, utils


if tf.__version__ >= '2.0': if tf.__version__ >= '2.0':


+ 4
- 4
modelscope/pipelines/multi_modal/image_captioning_pipeline.py View File

@@ -1,11 +1,11 @@
from typing import Any, Dict, Union
from typing import Any, Dict, Optional, Union


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.pipelines.base import Model, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import OfaImageCaptionPreprocessor, Preprocessor from modelscope.preprocessors import OfaImageCaptionPreprocessor, Preprocessor
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Model, Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()


@@ -16,7 +16,7 @@ class ImageCaptionPipeline(Pipeline):


def __init__(self, def __init__(self,
model: Union[Model, str], model: Union[Model, str],
preprocessor: [Preprocessor] = None,
preprocessor: Optional[Preprocessor] = None,
**kwargs): **kwargs):
""" """
use `model` and `preprocessor` to create a kws pipeline for prediction use `model` and `preprocessor` to create a kws pipeline for prediction


+ 2
- 3
modelscope/pipelines/multi_modal/multi_modal_embedding_pipeline.py View File

@@ -1,11 +1,10 @@
from typing import Any, Dict from typing import Any, Dict


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Model, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Model, Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 2
- 3
modelscope/pipelines/multi_modal/text_to_image_synthesis_pipeline.py View File

@@ -4,11 +4,10 @@ import torch


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input
from modelscope.pipelines.base import Input, Model, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from modelscope.utils.logger import get_logger from modelscope.utils.logger import get_logger
from ..base import Model, Pipeline
from ..builder import PIPELINES


logger = get_logger() logger = get_logger()




+ 7
- 7
modelscope/pipelines/multi_modal/visual_question_answering_pipeline.py View File

@@ -2,13 +2,13 @@ from typing import Any, Dict, Optional, Union


import torch import torch


from ...metainfo import Pipelines
from ...models import Model
from ...models.multi_modal import MPlugForVisualQuestionAnswering
from ...preprocessors import MPlugVisualQuestionAnsweringPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.multi_modal import MPlugForVisualQuestionAnswering
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import MPlugVisualQuestionAnsweringPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['VisualQuestionAnsweringPipeline'] __all__ = ['VisualQuestionAnsweringPipeline']




+ 8
- 8
modelscope/pipelines/nlp/dialog_intent_prediction_pipeline.py View File

@@ -2,14 +2,14 @@


from typing import Any, Dict, Union from typing import Any, Dict, Union


from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SpaceForDialogIntent
from ...outputs import OutputKeys
from ...preprocessors import DialogIntentPredictionPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline
from ..builder import PIPELINES
from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SpaceForDialogIntent
from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import DialogIntentPredictionPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['DialogIntentPredictionPipeline'] __all__ = ['DialogIntentPredictionPipeline']




+ 8
- 8
modelscope/pipelines/nlp/dialog_modeling_pipeline.py View File

@@ -2,14 +2,14 @@


from typing import Dict, Union from typing import Dict, Union


from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SpaceForDialogModeling
from ...outputs import OutputKeys
from ...preprocessors import DialogModelingPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SpaceForDialogModeling
from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import DialogModelingPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['DialogModelingPipeline'] __all__ = ['DialogModelingPipeline']




+ 7
- 7
modelscope/pipelines/nlp/dialog_state_tracking_pipeline.py View File

@@ -1,12 +1,12 @@
from typing import Any, Dict, Union from typing import Any, Dict, Union


from ...metainfo import Pipelines
from ...models import Model, SpaceForDialogStateTracking
from ...outputs import OutputKeys
from ...preprocessors import DialogStateTrackingPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline
from ..builder import PIPELINES
from modelscope.metainfo import Pipelines
from modelscope.models import Model, SpaceForDialogStateTracking
from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import DialogStateTrackingPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['DialogStateTrackingPipeline'] __all__ = ['DialogStateTrackingPipeline']




+ 8
- 8
modelscope/pipelines/nlp/fill_mask_pipeline.py View File

@@ -3,15 +3,15 @@ from typing import Any, Dict, Optional, Union


import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp.masked_language import MaskedLanguageModelBase
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp.masked_language import MaskedLanguageModelBase
from ...preprocessors import FillMaskPreprocessor
from ...utils.config import Config
from ...utils.constant import ModelFile, Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import FillMaskPreprocessor
from modelscope.utils.config import Config
from modelscope.utils.constant import ModelFile, Tasks


__all__ = ['FillMaskPipeline'] __all__ = ['FillMaskPipeline']
_type_map = {'veco': 'roberta', 'sbert': 'bert'} _type_map = {'veco': 'roberta', 'sbert': 'bert'}


+ 8
- 8
modelscope/pipelines/nlp/named_entity_recognition_pipeline.py View File

@@ -2,14 +2,14 @@ from typing import Any, Dict, Optional, Union


import torch import torch


from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import TransformerCRFForNamedEntityRecognition
from ...outputs import OutputKeys
from ...preprocessors import NERPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import TransformerCRFForNamedEntityRecognition
from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import NERPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['NamedEntityRecognitionPipeline'] __all__ = ['NamedEntityRecognitionPipeline']




+ 7
- 7
modelscope/pipelines/nlp/nli_pipeline.py View File

@@ -4,14 +4,14 @@ from typing import Any, Dict, Union
import numpy as np import numpy as np
import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SbertForNLI
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SbertForNLI
from ...preprocessors import NLIPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import NLIPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['NLIPipeline'] __all__ = ['NLIPipeline']




+ 7
- 7
modelscope/pipelines/nlp/sentence_similarity_pipeline.py View File

@@ -3,14 +3,14 @@ from typing import Any, Dict, Union
import numpy as np import numpy as np
import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SbertForSentenceSimilarity
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SbertForSentenceSimilarity
from ...preprocessors import SentenceSimilarityPreprocessor
from ...utils.constant import Tasks
from ..base import Input, Pipeline
from ..builder import PIPELINES
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import SentenceSimilarityPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['SentenceSimilarityPipeline'] __all__ = ['SentenceSimilarityPipeline']




+ 7
- 7
modelscope/pipelines/nlp/sentiment_classification_pipeline.py View File

@@ -3,14 +3,14 @@ from typing import Any, Dict, Union
import numpy as np import numpy as np
import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SequenceClassificationModel
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SequenceClassificationModel
from ...preprocessors import SentimentClassificationPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import SentimentClassificationPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['SentimentClassificationPipeline'] __all__ = ['SentimentClassificationPipeline']




+ 3
- 3
modelscope/pipelines/nlp/sequence_classification_pipeline.py View File

@@ -3,13 +3,13 @@ from typing import Any, Dict, Union
import numpy as np import numpy as np


from modelscope.metainfo import Pipelines from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import BertForSequenceClassification from modelscope.models.nlp import BertForSequenceClassification
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Input, Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import SequenceClassificationPreprocessor from modelscope.preprocessors import SequenceClassificationPreprocessor
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from ...models import Model
from ..base import Input, Pipeline
from ..builder import PIPELINES


__all__ = ['SequenceClassificationPipeline'] __all__ = ['SequenceClassificationPipeline']




+ 7
- 7
modelscope/pipelines/nlp/text_generation_pipeline.py View File

@@ -2,14 +2,14 @@ from typing import Any, Dict, Optional, Union


import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import PalmForTextGeneration
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import PalmForTextGeneration
from ...preprocessors import TextGenerationPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import TextGenerationPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['TextGenerationPipeline'] __all__ = ['TextGenerationPipeline']




+ 7
- 7
modelscope/pipelines/nlp/translation_pipeline.py View File

@@ -4,14 +4,14 @@ from typing import Any, Dict
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf


from modelscope.hub.snapshot_download import snapshot_download
from modelscope.metainfo import Pipelines
from modelscope.models.nlp import CsanmtForTranslation
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...hub.snapshot_download import snapshot_download
from ...metainfo import Pipelines
from ...models.nlp import CsanmtForTranslation
from ...utils.constant import ModelFile, Tasks
from ...utils.logger import get_logger
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.utils.constant import ModelFile, Tasks
from modelscope.utils.logger import get_logger


if tf.__version__ >= '2.0': if tf.__version__ >= '2.0':
tf = tf.compat.v1 tf = tf.compat.v1


+ 7
- 7
modelscope/pipelines/nlp/word_segmentation_pipeline.py View File

@@ -2,14 +2,14 @@ from typing import Any, Dict, Optional, Union


import torch import torch


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SbertForTokenClassification
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SbertForTokenClassification
from ...preprocessors import TokenClassificationPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline, Tensor
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import TokenClassificationPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['WordSegmentationPipeline'] __all__ = ['WordSegmentationPipeline']




+ 7
- 7
modelscope/pipelines/nlp/zero_shot_classification_pipeline.py View File

@@ -3,14 +3,14 @@ from typing import Any, Dict, Union
import torch import torch
from scipy.special import softmax from scipy.special import softmax


from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp import SbertForZeroShotClassification
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from ...metainfo import Pipelines
from ...models import Model
from ...models.nlp import SbertForZeroShotClassification
from ...preprocessors import ZeroShotClassificationPreprocessor
from ...utils.constant import Tasks
from ..base import Pipeline
from ..builder import PIPELINES
from modelscope.pipelines.base import Pipeline
from modelscope.pipelines.builder import PIPELINES
from modelscope.preprocessors import ZeroShotClassificationPreprocessor
from modelscope.utils.constant import Tasks


__all__ = ['ZeroShotClassificationPipeline'] __all__ = ['ZeroShotClassificationPipeline']




+ 5
- 5
modelscope/preprocessors/nlp.py View File

@@ -5,11 +5,11 @@ from typing import Any, Dict, Union


from transformers import AutoTokenizer from transformers import AutoTokenizer


from ..metainfo import Preprocessors
from ..models import Model
from ..utils.constant import Fields, InputFields, ModeKeys
from ..utils.hub import parse_label_mapping
from ..utils.type_assert import type_assert
from modelscope.metainfo import Preprocessors
from modelscope.models import Model
from modelscope.utils.constant import Fields, InputFields, ModeKeys
from modelscope.utils.hub import parse_label_mapping
from modelscope.utils.type_assert import type_assert
from .base import Preprocessor from .base import Preprocessor
from .builder import PREPROCESSORS from .builder import PREPROCESSORS




+ 2
- 3
modelscope/trainers/trainer.py View File

@@ -33,7 +33,7 @@ from modelscope.utils.logger import get_logger
from modelscope.utils.registry import build_from_cfg from modelscope.utils.registry import build_from_cfg
from modelscope.utils.tensor_utils import torch_default_data_collator from modelscope.utils.tensor_utils import torch_default_data_collator
from modelscope.utils.torch_utils import create_device, get_dist_info from modelscope.utils.torch_utils import create_device, get_dist_info
from modelscope.utils.utils import if_func_recieve_dict_inputs
from modelscope.utils.utils import if_func_receive_dict_inputs
from .base import BaseTrainer from .base import BaseTrainer
from .builder import TRAINERS from .builder import TRAINERS
from .default_config import DEFAULT_CONFIG from .default_config import DEFAULT_CONFIG
@@ -377,9 +377,8 @@ class EpochBasedTrainer(BaseTrainer):
model.train() model.train()
self._mode = ModeKeys.TRAIN self._mode = ModeKeys.TRAIN
inputs = self.collate_fn(inputs) inputs = self.collate_fn(inputs)

# call model forward but not __call__ to skip postprocess # call model forward but not __call__ to skip postprocess
if isinstance(inputs, Mapping) and not if_func_recieve_dict_inputs(
if isinstance(inputs, Mapping) and not if_func_receive_dict_inputs(
model.forward, inputs): model.forward, inputs):
train_outputs = model.forward(**inputs) train_outputs = model.forward(**inputs)
else: else:


+ 3
- 3
modelscope/trainers/utils/inference.py View File

@@ -13,7 +13,7 @@ from tqdm import tqdm


from modelscope.models.base import Model from modelscope.models.base import Model
from modelscope.utils.torch_utils import get_dist_info from modelscope.utils.torch_utils import get_dist_info
from modelscope.utils.utils import if_func_recieve_dict_inputs
from modelscope.utils.utils import if_func_receive_dict_inputs




def single_gpu_test(model, def single_gpu_test(model,
@@ -39,7 +39,7 @@ def single_gpu_test(model,
data = data_collate_fn(data) data = data_collate_fn(data)
with torch.no_grad(): with torch.no_grad():
if isinstance(data, if isinstance(data,
Mapping) and not if_func_recieve_dict_inputs(
Mapping) and not if_func_receive_dict_inputs(
model.forward, data): model.forward, data):


result = model(**data) result = model(**data)
@@ -93,7 +93,7 @@ def multi_gpu_test(model,
data = data_collate_fn(data) data = data_collate_fn(data)
with torch.no_grad(): with torch.no_grad():
if isinstance(data, if isinstance(data,
Mapping) and not if_func_recieve_dict_inputs(
Mapping) and not if_func_receive_dict_inputs(
model.forward, data): model.forward, data):
result = model(**data) result = model(**data)
else: else:


+ 1
- 1
modelscope/utils/tensor_utils.py View File

@@ -67,7 +67,7 @@ def torch_default_data_collator(features):
elif isinstance(v, list): elif isinstance(v, list):
batch[k] = torch.stack([d for f in features for d in f[k]]) batch[k] = torch.stack([d for f in features for d in f[k]])
else: else:
batch[k] = torch.tensor([f[k] for f in features])
batch[k] = torch.tensor(np.array([f[k] for f in features]))
elif isinstance(first, tuple): elif isinstance(first, tuple):
batch = [] batch = []
for idx in range(len(first)): for idx in range(len(first)):


+ 1
- 1
modelscope/utils/utils.py View File

@@ -3,7 +3,7 @@
import inspect import inspect




def if_func_recieve_dict_inputs(func, inputs):
def if_func_receive_dict_inputs(func, inputs):
"""to decide if a func could recieve dict inputs or not """to decide if a func could recieve dict inputs or not


Args: Args:


Loading…
Cancel
Save