├── .gitignore ├── LICENSE ├── README.md ├── celeba_dataset.py ├── celeba_u.py ├── celeba_u_script.ipynb ├── celeba_u_script.sh ├── checkpoints └── readme.txt ├── data ├── .gitignore └── readme.txt ├── distillate.py ├── images ├── artifacts.png ├── clipped_sampling.jpg └── no_clip.png ├── moving_average.py ├── prepare_data.py ├── requirements.txt ├── sample.py ├── strategies.py ├── train.py ├── train_utils.py ├── unet_ddpm.py └── v_diffusion.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/README.md -------------------------------------------------------------------------------- /celeba_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/celeba_dataset.py -------------------------------------------------------------------------------- /celeba_u.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/celeba_u.py -------------------------------------------------------------------------------- /celeba_u_script.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/celeba_u_script.ipynb -------------------------------------------------------------------------------- /celeba_u_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/celeba_u_script.sh -------------------------------------------------------------------------------- /checkpoints/readme.txt: -------------------------------------------------------------------------------- 1 | Put celeba folder with checkpoints here. -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | !readme.txt -------------------------------------------------------------------------------- /data/readme.txt: -------------------------------------------------------------------------------- 1 | Put celeba_256 here. -------------------------------------------------------------------------------- /distillate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/distillate.py -------------------------------------------------------------------------------- /images/artifacts.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/images/artifacts.png -------------------------------------------------------------------------------- /images/clipped_sampling.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/images/clipped_sampling.jpg -------------------------------------------------------------------------------- /images/no_clip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/images/no_clip.png -------------------------------------------------------------------------------- /moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/moving_average.py -------------------------------------------------------------------------------- /prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/prepare_data.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/sample.py -------------------------------------------------------------------------------- /strategies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/strategies.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/train.py -------------------------------------------------------------------------------- /train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/train_utils.py -------------------------------------------------------------------------------- /unet_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/unet_ddpm.py -------------------------------------------------------------------------------- /v_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hramchenko/diffusion_distiller/HEAD/v_diffusion.py --------------------------------------------------------------------------------