StableDiffusion/ldm/models/diffusion/__init__.py