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