ColossalAI/colossalai/legacy/pipeline/middleware/adaptor/__init__.py