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

6 lines
218 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
[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
[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
__all__ = ['CheckpointIO', 'CheckpointIndexFile', 'GeneralCheckpointIO']