├── .gitignore ├── LICENSE ├── README.md ├── imgs ├── outputs_imgs │ ├── 0.jpg │ └── 1.jpg └── test_imgs │ ├── cat.jpg │ └── cat_mask.jpg ├── ldm ├── data │ └── __init__.py ├── lr_scheduler.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ └── plms.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ └── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ └── test.png │ │ └── utils_image.py └── util.py ├── ldm_hacked ├── __init__.py ├── datasets.py └── logger.py ├── model_data └── sd_v15.yaml ├── predict.py ├── requirements.txt └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /imgs/outputs_imgs/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/imgs/outputs_imgs/0.jpg -------------------------------------------------------------------------------- /imgs/outputs_imgs/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/imgs/outputs_imgs/1.jpg -------------------------------------------------------------------------------- /imgs/test_imgs/cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/imgs/test_imgs/cat.jpg -------------------------------------------------------------------------------- /imgs/test_imgs/cat_mask.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/imgs/test_imgs/cat_mask.jpg -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm/util.py -------------------------------------------------------------------------------- /ldm_hacked/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm_hacked/__init__.py -------------------------------------------------------------------------------- /ldm_hacked/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm_hacked/datasets.py -------------------------------------------------------------------------------- /ldm_hacked/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/ldm_hacked/logger.py -------------------------------------------------------------------------------- /model_data/sd_v15.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/model_data/sd_v15.yaml -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/stable-diffusion/HEAD/train.py --------------------------------------------------------------------------------