ColossalAI/colossalai/legacy/inference/hybridengine/modeling/__init__.py