ColossalAI/colossalai/quantization/__init__.py