Files
ColossalAI/examples/images/diffusion/scripts/sample_diffusion.py