ColossalAI/examples/images/diffusion/ldm/data/__init__.py