github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/checkpoint_io/__init__.py

7 lines
318 B
Python
Raw Normal View History

[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
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
__all__ = ["CheckpointIO", "CheckpointIndexFile", "GeneralCheckpointIO", "HybridParallelCheckpointIO"]