File size: 1,433 Bytes
7feac49 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
# Copyright (c) Alibaba, Inc. and its affiliates.
from typing import TYPE_CHECKING
from swift.utils.import_utils import _LazyModule
if TYPE_CHECKING:
from .callback import extra_callbacks
from .loss import LOSS_MAPPING, get_loss_func
from .loss_scale import loss_scale_map
from .metric import InferStats, MeanMetric, Metric, compute_acc, get_metric, compute_rouge_bleu
from .optimizer import optimizers_map
from .agent_template import agent_templates
from .tuner import Tuner, extra_tuners, PeftTuner
from .prm import prms, PRM
from .orm import orms, ORM
from .multi_turn import multi_turns
from .rm_plugin import rm_plugins
else:
_import_structure = {
'callback': ['extra_callbacks'],
'loss': ['LOSS_MAPPING', 'get_loss_func'],
'loss_scale': ['loss_scale_map'],
'metric': ['InferStats', 'MeanMetric', 'Metric', 'compute_acc', 'get_metric', 'compute_rouge_bleu'],
'optimizer': ['optimizers_map'],
'agent_template': ['agent_templates'],
'tuner': ['Tuner', 'extra_tuners', 'PeftTuner'],
'prm': ['prms', 'PRM'],
'orm': ['orms', 'ORM'],
'multi_turn': ['multi_turns'],
'rm_plugin': ['rm_plugins']
}
import sys
sys.modules[__name__] = _LazyModule(
__name__,
globals()['__file__'],
_import_structure,
module_spec=__spec__,
extra_objects={},
)
|