StableDiffusion/ldm/modules/distributions/__init__.py