ColossalAI/colossalai/cli/__init__.py