ColossalAI/colossalai/checkpoint_io/__init__.py

from .checkpoint_io_base import CheckpointIO
from .general_checkpoint_io import GeneralCheckpointIO
from .hybrid_parallel_checkpoint_io import HybridParallelCheckpointIO
from .index_file import CheckpointIndexFile

__all__ = ['CheckpointIO', 'CheckpointIndexFile', 'GeneralCheckpointIO', 'HybridParallelCheckpointIO']
[checkpoint] refactored the API and added safetensors support (#3427) * [checkpoint] refactored the API and added safetensors support * polish code 2023-04-04 07:23:01 +00:00			`from .checkpoint_io_base import CheckpointIO`
[api] implemented the checkpoint io module (#3205) * [api] implemented the checkpoint io module * polish code * polish code 2023-03-23 02:53:17 +00:00			`from .general_checkpoint_io import GeneralCheckpointIO`
[hotfix] fix typo in hybrid parallel io (#4697) 2023-09-12 09:32:19 +00:00			`from .hybrid_parallel_checkpoint_io import HybridParallelCheckpointIO`
[checkpoint] refactored the API and added safetensors support (#3427) * [checkpoint] refactored the API and added safetensors support * polish code 2023-04-04 07:23:01 +00:00			`from .index_file import CheckpointIndexFile`
[api] implemented the checkpoint io module (#3205) * [api] implemented the checkpoint io module * polish code * polish code 2023-03-23 02:53:17 +00:00
[shardformer] support sharded checkpoint IO for models of HybridParallelPlugin (#4506) * add APIs * implement save_sharded_model * add test for hybrid checkpointio * implement naive loading for sharded model * implement efficient sharded model loading * open a new file for hybrid checkpoint_io * small fix * fix circular importing * fix docstring * arrange arguments and apis * small fix 2023-08-25 14:04:57 +00:00			`__all__ = ['CheckpointIO', 'CheckpointIndexFile', 'GeneralCheckpointIO', 'HybridParallelCheckpointIO']`