├── .gitignore ├── README.md ├── ddpm.py ├── dit_model.py ├── ema.png ├── inference.py ├── main.py └── run.sh /.gitignore: -------------------------------------------------------------------------------- 1 | wandb 2 | local_train_dir 3 | ckpt 4 | __pycache__/ 5 | *_image.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/README.md -------------------------------------------------------------------------------- /ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/ddpm.py -------------------------------------------------------------------------------- /dit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/dit_model.py -------------------------------------------------------------------------------- /ema.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/ema.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/inference.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/main.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cloneofsimo/min-max-in-dit/HEAD/run.sh --------------------------------------------------------------------------------