11 lines
395 B
Python
11 lines
395 B
Python
# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
from modelscope.utils.import_utils import is_torch_available
|
|
from .base_head import * # noqa F403
|
|
from .base_model import * # noqa F403
|
|
|
|
if is_torch_available():
|
|
from .base_torch_model import TorchModel
|
|
from .base_torch_head import TorchHead
|