xtuner/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
import os

from mmengine.utils import digit_version

from .entry_point import cli
from .version import __version__, version_info

HF_CEPH_HUB = os.getenv('HF_CEPH_HUB', '')
HF_USE_CEPH = os.getenv('HF_USE_CEPH', 0) or HF_CEPH_HUB != ''
DS_CEPH_DIR = os.getenv('DS_CEPH_DIR', None)
if HF_USE_CEPH:
    from .utils.fileio import (patch_hf_auto_from_pretrained,
                               patch_hf_save_pretrained)
    patch_hf_auto_from_pretrained(HF_CEPH_HUB)
    patch_hf_save_pretrained()

if DS_CEPH_DIR:
    from .utils.fileio import patch_deepspeed_engine
    patch_deepspeed_engine()

__all__ = [
    '__version__', 'version_info', 'digit_version', 'cli', 'HF_USE_CEPH',
    'DS_CEPH_DIR'
]