mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 12:39:25 +01:00
96 lines
4.2 KiB
Python
96 lines
4.2 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
from typing import TYPE_CHECKING
|
|
|
|
from modelscope.utils.import_utils import LazyImportModule
|
|
|
|
if TYPE_CHECKING:
|
|
from .exporters import Exporter, TfModelExporter, TorchModelExporter
|
|
from .hub.api import HubApi
|
|
from .hub.check_model import check_local_model_is_latest, check_model_is_id
|
|
from .hub.push_to_hub import push_to_hub, push_to_hub_async
|
|
from .hub.snapshot_download import snapshot_download
|
|
from .metrics import (
|
|
AccuracyMetric, AudioNoiseMetric, BleuMetric, ImageColorEnhanceMetric,
|
|
ImageColorizationMetric, ImageDenoiseMetric, ImageInpaintingMetric,
|
|
ImageInstanceSegmentationCOCOMetric, ImagePortraitEnhancementMetric,
|
|
ImageQualityAssessmentDegradationMetric,
|
|
ImageQualityAssessmentMosMetric, LossMetric, Metric,
|
|
MovieSceneSegmentationMetric, OCRRecognitionMetric, PplMetric,
|
|
ReferringVideoObjectSegmentationMetric, SequenceClassificationMetric,
|
|
TextGenerationMetric, TextRankingMetric, TokenClassificationMetric,
|
|
VideoFrameInterpolationMetric, VideoStabilizationMetric,
|
|
VideoSummarizationMetric, VideoSuperResolutionMetric,
|
|
task_default_metrics)
|
|
from .models import Model, TorchModel
|
|
from .msdatasets import MsDataset
|
|
from .pipelines import Pipeline, pipeline
|
|
from .preprocessors import Preprocessor
|
|
from .trainers import (EpochBasedTrainer, Hook, Priority, TrainingArgs,
|
|
build_dataset_from_file)
|
|
from .utils.constant import Tasks
|
|
from .utils.hf_util import (AutoConfig, AutoModel, AutoModelForCausalLM,
|
|
AutoModelForSeq2SeqLM,
|
|
AutoModelForSequenceClassification,
|
|
AutoModelForTokenClassification, AutoTokenizer,
|
|
GenerationConfig)
|
|
from .utils.hub import create_model_if_not_exist, read_config
|
|
from .utils.logger import get_logger
|
|
from .version import __release_datetime__, __version__
|
|
|
|
else:
|
|
_import_structure = {
|
|
'version': ['__release_datetime__', '__version__'],
|
|
'trainers': [
|
|
'EpochBasedTrainer', 'TrainingArgs', 'Hook', 'Priority',
|
|
'build_dataset_from_file'
|
|
],
|
|
'exporters': [
|
|
'Exporter',
|
|
'TfModelExporter',
|
|
'TorchModelExporter',
|
|
],
|
|
'hub.api': ['HubApi'],
|
|
'hub.snapshot_download': ['snapshot_download'],
|
|
'hub.push_to_hub': ['push_to_hub', 'push_to_hub_async'],
|
|
'hub.check_model':
|
|
['check_model_is_id', 'check_local_model_is_latest'],
|
|
'metrics': [
|
|
'AudioNoiseMetric', 'Metric', 'task_default_metrics',
|
|
'ImageColorEnhanceMetric', 'ImageDenoiseMetric',
|
|
'ImageInstanceSegmentationCOCOMetric',
|
|
'ImagePortraitEnhancementMetric', 'SequenceClassificationMetric',
|
|
'TextGenerationMetric', 'TokenClassificationMetric',
|
|
'VideoSummarizationMetric', 'MovieSceneSegmentationMetric',
|
|
'AccuracyMetric', 'BleuMetric', 'ImageInpaintingMetric',
|
|
'ReferringVideoObjectSegmentationMetric',
|
|
'VideoFrameInterpolationMetric', 'VideoStabilizationMetric',
|
|
'VideoSuperResolutionMetric', 'PplMetric',
|
|
'ImageQualityAssessmentDegradationMetric',
|
|
'ImageQualityAssessmentMosMetric', 'TextRankingMetric',
|
|
'LossMetric', 'ImageColorizationMetric', 'OCRRecognitionMetric'
|
|
],
|
|
'models': ['Model', 'TorchModel'],
|
|
'preprocessors': ['Preprocessor'],
|
|
'pipelines': ['Pipeline', 'pipeline'],
|
|
'utils.hub': ['read_config', 'create_model_if_not_exist'],
|
|
'utils.logger': ['get_logger'],
|
|
'utils.constant': ['Tasks'],
|
|
'utils.hf_util': [
|
|
'AutoConfig', 'GenerationConfig', 'AutoModel',
|
|
'AutoModelForCausalLM', 'AutoModelForSeq2SeqLM', 'AutoTokenizer',
|
|
'AutoModelForSequenceClassification',
|
|
'AutoModelForTokenClassification'
|
|
],
|
|
'msdatasets': ['MsDataset']
|
|
}
|
|
|
|
import sys
|
|
|
|
sys.modules[__name__] = LazyImportModule(
|
|
__name__,
|
|
globals()['__file__'],
|
|
_import_structure,
|
|
module_spec=__spec__,
|
|
extra_objects={},
|
|
)
|