| # 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' | |
| ] | |