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