Files
ColossalAI/examples/images/diffusion/ldm/models/autoencoder.py