├── .dockerignore ├── .gitignore ├── LDSR.py ├── LICENSE ├── README.md ├── README_LD.md ├── __init__.py ├── assets ├── a-painting-of-a-fire.png ├── a-photograph-of-a-fire.png ├── a-shirt-with-a-fire-printed-on-it.png ├── a-shirt-with-the-inscription-'fire'.png ├── a-watercolor-painting-of-a-fire.png ├── birdhouse.png ├── fire.png ├── inpainting.png ├── modelfigure.png ├── reconstruction1.png ├── reconstruction2.png ├── results.gif ├── the-earth-is-on-fire,-oil-on-canvas.png ├── txt2img-convsample.png └── txt2img-preview.png ├── batch.py ├── cog.yaml ├── cogflare.py ├── configs ├── autoencoder │ ├── autoencoder_kl_16x16x16.yaml │ ├── autoencoder_kl_32x32x4.yaml │ ├── autoencoder_kl_64x64x3.yaml │ └── autoencoder_kl_8x8x64.yaml └── latent-diffusion │ ├── celebahq-ldm-vq-4.yaml │ ├── cin-ldm-vq-f8.yaml │ ├── cin256-v2.yaml │ ├── ffhq-ldm-vq-4.yaml │ ├── lsun_bedrooms-ldm-vq-4.yaml │ ├── lsun_churches-ldm-kl-8.yaml │ └── txt2img-1p4B-eval.yaml ├── data ├── DejaVuSans.ttf ├── example_conditioning │ ├── superresolution │ │ └── sample_0.jpg │ └── text_conditional │ │ └── sample_0.txt ├── imagenet_clsidx_to_label.txt ├── imagenet_train_hr_indices.p ├── imagenet_val_hr_indices.p ├── index_synset.yaml └── inpainting_examples │ ├── 6458524847_2f4c361183_k.png │ ├── 6458524847_2f4c361183_k_mask.png │ ├── 8399166846_f6fb4e4b8e_k.png │ ├── 8399166846_f6fb4e4b8e_k_mask.png │ ├── alex-iby-G_Pk4D9rMLs.png │ ├── alex-iby-G_Pk4D9rMLs_mask.png │ ├── bench2.png │ ├── bench2_mask.png │ ├── bertrand-gabioud-CpuFzIsHYJ0.png │ ├── bertrand-gabioud-CpuFzIsHYJ0_mask.png │ ├── billow926-12-Wc-Zgx6Y.png │ ├── billow926-12-Wc-Zgx6Y_mask.png │ ├── overture-creations-5sI6fQgYIuo.png │ ├── overture-creations-5sI6fQgYIuo_mask.png │ ├── photo-1583445095369-9c651e7e5d34.png │ └── photo-1583445095369-9c651e7e5d34_mask.png ├── download_model.bat ├── environment.yaml ├── ldm ├── data │ ├── __init__.py │ ├── base.py │ ├── imagenet.py │ └── lsun.py ├── models │ └── diffusion │ │ ├── __init__.py │ │ └── ddim.py ├── modules │ └── diffusionmodules │ │ ├── __init__.py │ │ └── util.py └── util.py ├── main.py ├── requirements.txt └── setup.py /.dockerignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LDSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/LDSR.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /README_LD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/README_LD.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /assets/a-painting-of-a-fire.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/a-painting-of-a-fire.png -------------------------------------------------------------------------------- /assets/a-photograph-of-a-fire.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/a-photograph-of-a-fire.png -------------------------------------------------------------------------------- /assets/a-shirt-with-a-fire-printed-on-it.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/a-shirt-with-a-fire-printed-on-it.png -------------------------------------------------------------------------------- /assets/a-shirt-with-the-inscription-'fire'.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/a-shirt-with-the-inscription-'fire'.png -------------------------------------------------------------------------------- /assets/a-watercolor-painting-of-a-fire.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/a-watercolor-painting-of-a-fire.png -------------------------------------------------------------------------------- /assets/birdhouse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/birdhouse.png -------------------------------------------------------------------------------- /assets/fire.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/fire.png -------------------------------------------------------------------------------- /assets/inpainting.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/inpainting.png -------------------------------------------------------------------------------- /assets/modelfigure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/modelfigure.png -------------------------------------------------------------------------------- /assets/reconstruction1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/reconstruction1.png -------------------------------------------------------------------------------- /assets/reconstruction2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/reconstruction2.png -------------------------------------------------------------------------------- /assets/results.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/results.gif -------------------------------------------------------------------------------- /assets/the-earth-is-on-fire,-oil-on-canvas.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/the-earth-is-on-fire,-oil-on-canvas.png -------------------------------------------------------------------------------- /assets/txt2img-convsample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/txt2img-convsample.png -------------------------------------------------------------------------------- /assets/txt2img-preview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/assets/txt2img-preview.png -------------------------------------------------------------------------------- /batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/batch.py -------------------------------------------------------------------------------- /cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/cog.yaml -------------------------------------------------------------------------------- /cogflare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/cogflare.py -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/celebahq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/celebahq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/cin-ldm-vq-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/cin-ldm-vq-f8.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/cin256-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/cin256-v2.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/ffhq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/ffhq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-eval.yaml -------------------------------------------------------------------------------- /data/DejaVuSans.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/DejaVuSans.ttf -------------------------------------------------------------------------------- /data/example_conditioning/superresolution/sample_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/example_conditioning/superresolution/sample_0.jpg -------------------------------------------------------------------------------- /data/example_conditioning/text_conditional/sample_0.txt: -------------------------------------------------------------------------------- 1 | A basket of cerries 2 | -------------------------------------------------------------------------------- /data/imagenet_clsidx_to_label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/imagenet_clsidx_to_label.txt -------------------------------------------------------------------------------- /data/imagenet_train_hr_indices.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/imagenet_train_hr_indices.p -------------------------------------------------------------------------------- /data/imagenet_val_hr_indices.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/imagenet_val_hr_indices.p -------------------------------------------------------------------------------- /data/index_synset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/index_synset.yaml -------------------------------------------------------------------------------- /data/inpainting_examples/6458524847_2f4c361183_k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/6458524847_2f4c361183_k.png -------------------------------------------------------------------------------- /data/inpainting_examples/6458524847_2f4c361183_k_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/6458524847_2f4c361183_k_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/8399166846_f6fb4e4b8e_k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/8399166846_f6fb4e4b8e_k.png -------------------------------------------------------------------------------- /data/inpainting_examples/8399166846_f6fb4e4b8e_k_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/8399166846_f6fb4e4b8e_k_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/alex-iby-G_Pk4D9rMLs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/alex-iby-G_Pk4D9rMLs.png -------------------------------------------------------------------------------- /data/inpainting_examples/alex-iby-G_Pk4D9rMLs_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/alex-iby-G_Pk4D9rMLs_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/bench2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/bench2.png -------------------------------------------------------------------------------- /data/inpainting_examples/bench2_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/bench2_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/bertrand-gabioud-CpuFzIsHYJ0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/bertrand-gabioud-CpuFzIsHYJ0.png -------------------------------------------------------------------------------- /data/inpainting_examples/bertrand-gabioud-CpuFzIsHYJ0_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/bertrand-gabioud-CpuFzIsHYJ0_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/billow926-12-Wc-Zgx6Y.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/billow926-12-Wc-Zgx6Y.png -------------------------------------------------------------------------------- /data/inpainting_examples/billow926-12-Wc-Zgx6Y_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/billow926-12-Wc-Zgx6Y_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/overture-creations-5sI6fQgYIuo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png -------------------------------------------------------------------------------- /data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png -------------------------------------------------------------------------------- /data/inpainting_examples/photo-1583445095369-9c651e7e5d34.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/photo-1583445095369-9c651e7e5d34.png -------------------------------------------------------------------------------- /data/inpainting_examples/photo-1583445095369-9c651e7e5d34_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/data/inpainting_examples/photo-1583445095369-9c651e7e5d34_mask.png -------------------------------------------------------------------------------- /download_model.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/download_model.bat -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/environment.yaml -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/data/imagenet.py -------------------------------------------------------------------------------- /ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/data/lsun.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/ldm/util.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hafiidz/latent-diffusion/HEAD/setup.py --------------------------------------------------------------------------------