toolkits/distributed_checkpoints_convertor/impl/general/__init__.py (6 lines of code) (raw):

from .h2m_synchronizer import HF2MGSynchronizer from .m2h_synchronizer import MG2HFSynchronizer __all__ = [ 'HF2MGSynchronizer', 'MG2HFSynchronizer' ]