├── .github └── workflow │ └── auto_assign.yml ├── LICENSE ├── README.md ├── environment.yml ├── experiments-results-analysis └── experiments │ ├── add_dog8_config_01.sh │ ├── add_dog8_config_01.yaml │ ├── replace_dog8_config_01.sh │ └── replace_dog8_config_01.yaml ├── requirements.txt ├── setup.cfg ├── setup.py └── src ├── __init__.py ├── auto_evaluation.py ├── clip_img_ret.py ├── configs ├── autoencoder │ ├── autoencoder_kl_16x16x16.yaml │ ├── autoencoder_kl_32x32x4.yaml │ ├── autoencoder_kl_64x64x3.yaml │ └── autoencoder_kl_8x8x64.yaml ├── dream-edit │ ├── add_dog_default.yaml │ └── edit_backpack_default.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 ├── generate_new.py ├── iterate_generate.py ├── ldm ├── __init__.py ├── data │ ├── __init__.py │ ├── base.py │ ├── imagenet.py │ └── lsun.py ├── lr_scheduler.py ├── models │ ├── __init__.py │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ └── plms.py ├── modules │ ├── __init__.py │ ├── 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 │ │ │ └── test.png │ │ └── utils_image.py │ ├── losses │ │ ├── __init__.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ └── x_transformer.py └── util.py ├── metrics ├── __init__.py ├── clip_vit.py ├── dino_vit.py ├── distances.py └── evaluate_dino.py ├── pipelines ├── __init__.py ├── extract_object_pipeline.py ├── imagecaption_pipelines.py └── inpainting_pipelines.py └── utils ├── losses.py ├── mask_helper.py ├── path_finder.py └── visual_helper.py /.github/workflow/auto_assign.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/.github/workflow/auto_assign.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/environment.yml -------------------------------------------------------------------------------- /experiments-results-analysis /experiments/add_dog8_config_01.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/experiments-results-analysis /experiments/add_dog8_config_01.sh -------------------------------------------------------------------------------- /experiments-results-analysis /experiments/add_dog8_config_01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/experiments-results-analysis /experiments/add_dog8_config_01.yaml -------------------------------------------------------------------------------- /experiments-results-analysis /experiments/replace_dog8_config_01.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/experiments-results-analysis /experiments/replace_dog8_config_01.sh -------------------------------------------------------------------------------- /experiments-results-analysis /experiments/replace_dog8_config_01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/experiments-results-analysis /experiments/replace_dog8_config_01.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/auto_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/auto_evaluation.py -------------------------------------------------------------------------------- /src/clip_img_ret.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/clip_img_ret.py -------------------------------------------------------------------------------- /src/configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /src/configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /src/configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /src/configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /src/configs/dream-edit/add_dog_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/dream-edit/add_dog_default.yaml -------------------------------------------------------------------------------- /src/configs/dream-edit/edit_backpack_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/dream-edit/edit_backpack_default.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/celebahq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/celebahq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/cin-ldm-vq-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/cin-ldm-vq-f8.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/cin256-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/cin256-v2.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/ffhq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/ffhq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml -------------------------------------------------------------------------------- /src/configs/latent-diffusion/txt2img-1p4B-eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/latent-diffusion/txt2img-1p4B-eval.yaml -------------------------------------------------------------------------------- /src/configs/retrieval-augmented-diffusion/768x768.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/retrieval-augmented-diffusion/768x768.yaml -------------------------------------------------------------------------------- /src/configs/stable-diffusion/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/configs/stable-diffusion/v1-inference.yaml -------------------------------------------------------------------------------- /src/generate_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/generate_new.py -------------------------------------------------------------------------------- /src/iterate_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/iterate_generate.py -------------------------------------------------------------------------------- /src/ldm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/data/base.py -------------------------------------------------------------------------------- /src/ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/data/imagenet.py -------------------------------------------------------------------------------- /src/ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/data/lsun.py -------------------------------------------------------------------------------- /src/ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /src/ldm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /src/ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /src/ldm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/attention.py -------------------------------------------------------------------------------- /src/ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /src/ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /src/ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /src/ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /src/ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/ema.py -------------------------------------------------------------------------------- /src/ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /src/ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /src/ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /src/ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /src/ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /src/ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /src/ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /src/ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /src/ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /src/ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /src/ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/ldm/util.py -------------------------------------------------------------------------------- /src/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/metrics/clip_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/metrics/clip_vit.py -------------------------------------------------------------------------------- /src/metrics/dino_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/metrics/dino_vit.py -------------------------------------------------------------------------------- /src/metrics/distances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/metrics/distances.py -------------------------------------------------------------------------------- /src/metrics/evaluate_dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/metrics/evaluate_dino.py -------------------------------------------------------------------------------- /src/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pipelines/extract_object_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/pipelines/extract_object_pipeline.py -------------------------------------------------------------------------------- /src/pipelines/imagecaption_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/pipelines/imagecaption_pipelines.py -------------------------------------------------------------------------------- /src/pipelines/inpainting_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/pipelines/inpainting_pipelines.py -------------------------------------------------------------------------------- /src/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/utils/losses.py -------------------------------------------------------------------------------- /src/utils/mask_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/utils/mask_helper.py -------------------------------------------------------------------------------- /src/utils/path_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/utils/path_finder.py -------------------------------------------------------------------------------- /src/utils/visual_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DreamEditBenchTeam/DreamEdit/HEAD/src/utils/visual_helper.py --------------------------------------------------------------------------------