Files
ColossalAI/colossalai/utils/checkpointing.py