Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9491629master
| @@ -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) | ||||
| @@ -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) | ||||
| @@ -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 | ||||
| @@ -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, | ||||
| @@ -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') | ||||
| @@ -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 | ||||
| @@ -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 | ||||
| @@ -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 | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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 | ||||
| @@ -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 | ||||
| @@ -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): | ||||
| @@ -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__) | ||||
| @@ -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__) | ||||
| @@ -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,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'] | ||||
| @@ -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,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( | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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,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,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 | ||||
| @@ -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): | ||||
| @@ -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'] | ||||
| @@ -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,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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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) | ||||
| @@ -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): | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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 | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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': | ||||
| @@ -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 | ||||
| @@ -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() | ||||
| @@ -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() | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'} | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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'] | ||||
| @@ -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,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'] | ||||
| @@ -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'] | ||||
| @@ -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 | ||||
| @@ -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'] | ||||
| @@ -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,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 | ||||
| @@ -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: | ||||
| @@ -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: | ||||
| @@ -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)): | ||||
| @@ -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: | ||||