├── LICENSE ├── README.md ├── augmentation.py ├── configs ├── pixel_diffusion.yaml └── pixel_diffusion_a.yaml ├── ema_model.py ├── generate.py ├── model.py ├── model_a.py ├── pipeline.py ├── requirements.txt ├── scripts └── solve_posthoc_ema.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/README.md -------------------------------------------------------------------------------- /augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/augmentation.py -------------------------------------------------------------------------------- /configs/pixel_diffusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/configs/pixel_diffusion.yaml -------------------------------------------------------------------------------- /configs/pixel_diffusion_a.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/configs/pixel_diffusion_a.yaml -------------------------------------------------------------------------------- /ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/ema_model.py -------------------------------------------------------------------------------- /generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/generate.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/model.py -------------------------------------------------------------------------------- /model_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/model_a.py -------------------------------------------------------------------------------- /pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/pipeline.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/solve_posthoc_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/scripts/solve_posthoc_ema.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mmathew23/improved_edm/HEAD/train.py --------------------------------------------------------------------------------