├── .gitignore ├── LICENSE ├── README.md ├── multi-scale-blended-diffusion ├── InteractiveEditing.ipynb ├── OrignalLatentDiffusionLICENSE ├── 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 │ ├── retrieval-augmented-diffusion │ │ └── 768x768.yaml │ └── stable-diffusion │ │ └── v1-inference.yaml ├── environment.yaml ├── inputs │ ├── bedroom_painting.jpg │ ├── clarissa_strozzi.jpg │ ├── inputs.txt │ ├── lofi_revoy.jpg │ ├── prof_pic.jpg │ └── readme.md ├── ldm │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── base.py │ │ ├── imagenet.py │ │ └── lsun.py │ ├── lr_scheduler.py │ ├── models │ │ ├── autoencoder.py │ │ └── diffusion │ │ │ ├── __init__.py │ │ │ ├── classifier.py │ │ │ ├── ddim.py │ │ │ ├── ddpm.py │ │ │ └── plms.py │ ├── modules │ │ ├── attention.py │ │ ├── diffusionmodules │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ ├── openaimodel.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_image.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── contperceptual.py │ │ │ └── vqperceptual.py │ │ └── x_transformer.py │ └── util.py ├── msbd │ ├── BLDSampler.py │ ├── MSBDGenerator.py │ ├── __init__.py │ └── msbd_utils.py └── multi_scale_blended_diffusion.py └── overview.jpg /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/InteractiveEditing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/InteractiveEditing.ipynb -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/OrignalLatentDiffusionLICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/OrignalLatentDiffusionLICENSE -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/celebahq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/celebahq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/cin-ldm-vq-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/cin-ldm-vq-f8.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/cin256-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/cin256-v2.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/ffhq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/ffhq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/retrieval-augmented-diffusion/768x768.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/retrieval-augmented-diffusion/768x768.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/configs/stable-diffusion/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/configs/stable-diffusion/v1-inference.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/environment.yaml -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/bedroom_painting.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/bedroom_painting.jpg -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/clarissa_strozzi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/clarissa_strozzi.jpg -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/inputs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/inputs.txt -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/lofi_revoy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/lofi_revoy.jpg -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/prof_pic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/prof_pic.jpg -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/inputs/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/inputs/readme.md -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/data/base.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/data/imagenet.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/data/lsun.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/attention.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/ema.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/ldm/util.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/msbd/BLDSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/msbd/BLDSampler.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/msbd/MSBDGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/msbd/MSBDGenerator.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/msbd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/msbd/msbd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/msbd/msbd_utils.py -------------------------------------------------------------------------------- /multi-scale-blended-diffusion/multi_scale_blended_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/multi-scale-blended-diffusion/multi_scale_blended_diffusion.py -------------------------------------------------------------------------------- /overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pfnet-research/multi-stage-blended-diffusion/HEAD/overview.jpg --------------------------------------------------------------------------------