mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-02 12:09:36 +02:00
4dca4773db
1. Support csanmt exporting to savedmodel format 2. Create a new base class for text-ranking preprocessors, and move some parameters of mgeo_ranking_preprocessor to init method 3. Avoid Model & Preprocessor classes coupled with pytorch 4. Regression test supports comparing only model output 5. Support zero-shot exporting to onnx and torchscript Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11522461
14 lines
519 B
Python
14 lines
519 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
from typing import TYPE_CHECKING
|
|
|
|
from modelscope.utils.error import (AUDIO_IMPORT_ERROR,
|
|
TENSORFLOW_IMPORT_WARNING)
|
|
from modelscope.utils.import_utils import is_torch_available
|
|
from . import audio, cv, multi_modal, nlp
|
|
from .base import Head, Model
|
|
from .builder import BACKBONES, HEADS, MODELS, build_model
|
|
|
|
if is_torch_available():
|
|
from .base.base_torch_model import TorchModel
|
|
from .base.base_torch_head import TorchHead
|