mirror of https://github.com/hpcaitech/ColossalAI
1beb85cc25
* [checkpoint] refactored the API and added safetensors support * polish code |
||
|---|---|---|
| .. | ||
| components_to_test | ||
| kit | ||
| test_amp | ||
| test_analyzer | ||
| test_auto_parallel | ||
| test_autochunk | ||
| test_booster | ||
| test_checkpoint_io | ||
| test_cluster | ||
| test_comm | ||
| test_config | ||
| test_context | ||
| test_data | ||
| test_data_pipeline_tensor_parallel | ||
| test_ddp | ||
| test_device | ||
| test_engine | ||
| test_fx | ||
| test_gemini | ||
| test_layers | ||
| test_moe | ||
| test_ops | ||
| test_optimizer | ||
| test_pipeline | ||
| test_tensor | ||
| test_trainer | ||
| test_utils | ||
| test_zero | ||
| __init__.py | ||