xtuner/utils/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .constants import (DEFAULT_IMAGE_TOKEN, DEFAULT_PAD_TOKEN_INDEX,
                        IGNORE_INDEX, IMAGE_TOKEN_INDEX)
from .handle_moe_load_and_save import (SUPPORT_MODELS, get_origin_state_dict,
                                       load_state_dict_into_model)
from .stop_criteria import StopWordStoppingCriteria
from .templates import PROMPT_TEMPLATE, SYSTEM_TEMPLATE

__all__ = [
    'IGNORE_INDEX', 'DEFAULT_PAD_TOKEN_INDEX', 'PROMPT_TEMPLATE',
    'DEFAULT_IMAGE_TOKEN', 'SYSTEM_TEMPLATE', 'StopWordStoppingCriteria',
    'IMAGE_TOKEN_INDEX', 'load_state_dict_into_model', 'get_origin_state_dict',
    'SUPPORT_MODELS'
]