Files
ColossalAI/colossalai/utils/model/colo_init_context.py