ColossalAI/tests/kit/model_zoo/torchaudio/__init__.py