├── .gitignore ├── LICENSE ├── README.md ├── assets └── Figures │ ├── Gradio.png │ ├── Teaser.png │ ├── demo_gradio_editing.png │ ├── demo_gradio_generation.png │ └── demo_gradio_ops.png ├── config └── ranni_sdv21_v1.yaml ├── demo_gradio.py ├── environment.yaml ├── ldm ├── data │ ├── __init__.py │ └── util.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ ├── plms.py │ │ └── sampling_util.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 │ ├── karlo │ │ ├── __init__.py │ │ ├── diffusers_pipeline.py │ │ └── kakao │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── clip.py │ │ │ ├── decoder_model.py │ │ │ ├── prior_model.py │ │ │ ├── sr_256_1k.py │ │ │ └── sr_64_256.py │ │ │ ├── modules │ │ │ ├── __init__.py │ │ │ ├── diffusion │ │ │ │ ├── gaussian_diffusion.py │ │ │ │ └── respace.py │ │ │ ├── nn.py │ │ │ ├── resample.py │ │ │ ├── unet.py │ │ │ └── xf.py │ │ │ ├── sampler.py │ │ │ └── template.py │ └── midas │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── blocks.py │ │ ├── dpt_depth.py │ │ ├── midas_net.py │ │ ├── midas_net_custom.py │ │ ├── transforms.py │ │ └── vit.py │ │ └── utils.py └── util.py ├── ranni ├── ddim_hacked.py ├── hack.py ├── llama_configuration.py ├── llama_modeling.py ├── panel_conditioner.py └── ranni.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/README.md -------------------------------------------------------------------------------- /assets/Figures/Gradio.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/assets/Figures/Gradio.png -------------------------------------------------------------------------------- /assets/Figures/Teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/assets/Figures/Teaser.png -------------------------------------------------------------------------------- /assets/Figures/demo_gradio_editing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/assets/Figures/demo_gradio_editing.png -------------------------------------------------------------------------------- /assets/Figures/demo_gradio_generation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/assets/Figures/demo_gradio_generation.png -------------------------------------------------------------------------------- /assets/Figures/demo_gradio_ops.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/assets/Figures/demo_gradio_ops.png -------------------------------------------------------------------------------- /config/ranni_sdv21_v1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/config/ranni_sdv21_v1.yaml -------------------------------------------------------------------------------- /demo_gradio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/demo_gradio.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/environment.yaml -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/data/util.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/karlo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/karlo/diffusers_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/diffusers_pipeline.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/models/clip.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/decoder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/models/decoder_model.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/prior_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/models/prior_model.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/sr_256_1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/models/sr_256_1k.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/models/sr_64_256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/models/sr_64_256.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/__init__.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/diffusion/respace.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/nn.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/resample.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/unet.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/modules/xf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/modules/xf.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/sampler.py -------------------------------------------------------------------------------- /ldm/modules/karlo/kakao/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/karlo/kakao/template.py -------------------------------------------------------------------------------- /ldm/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/api.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /ldm/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/modules/midas/utils.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ldm/util.py -------------------------------------------------------------------------------- /ranni/ddim_hacked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/ddim_hacked.py -------------------------------------------------------------------------------- /ranni/hack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/hack.py -------------------------------------------------------------------------------- /ranni/llama_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/llama_configuration.py -------------------------------------------------------------------------------- /ranni/llama_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/llama_modeling.py -------------------------------------------------------------------------------- /ranni/panel_conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/panel_conditioner.py -------------------------------------------------------------------------------- /ranni/ranni.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/ranni/ranni.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ali-vilab/Ranni/HEAD/utils.py --------------------------------------------------------------------------------