mirror of https://github.com/hpcaitech/ColossalAI
![]() * hybrid plugin support huggingface from_pretrained * add huggingface compatibility tests * add folder cleaning * fix bugs |
||
---|---|---|
.. | ||
test_gemini_checkpoint_io.py | ||
test_gemini_torch_compability.py | ||
test_general_checkpoint_io.py | ||
test_hybrid_huggingface_compatibility.py | ||
test_hybrid_parallel_plugin_checkpoint_io.py | ||
test_low_level_zero_checkpoint_io.py | ||
test_torch_ddp_checkpoint_io.py | ||
test_torch_fsdp_checkpoint_io.py | ||
utils.py |