ColossalAI

Author	SHA1	Message	Date
Baizhou Zhang	d8ceeac14e	[hotfix] fix typo in hybrid parallel io (#4697 )	2023-09-12 17:32:19 +08:00
Baizhou Zhang	44eab2b27f	[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 22:04:57 +08:00
Frank Lee	1beb85cc25	[checkpoint] refactored the API and added safetensors support (#3427 ) * [checkpoint] refactored the API and added safetensors support * polish code	2023-04-04 15:23:01 +08:00
Frank Lee	cd142fbefa	[api] implemented the checkpoint io module (#3205 ) * [api] implemented the checkpoint io module * polish code * polish code	2023-03-23 10:53:17 +08:00