├── .gitignore ├── LICENSE.txt ├── README.md ├── diffusion ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── diffusion_utils.cpython-39.pyc │ ├── gaussian_diffusion.cpython-39.pyc │ └── respace.cpython-39.pyc ├── diffusion_utils.py ├── gaussian_diffusion.py ├── respace.py └── timestep_sampler.py ├── download.py ├── evaluator.py ├── imgs ├── effect.png └── scheme.png ├── models.py ├── requirements.txt ├── sample_ddp.py ├── train.py ├── train_accelerate.py ├── udit_models.py └── utils ├── fastdit_kit.py ├── parser_setter.py ├── test_latency_throughout.py └── udit_models_original.py /.gitignore: -------------------------------------------------------------------------------- 1 | pt 2 | results 3 | samples -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/README.md -------------------------------------------------------------------------------- /diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/__init__.py -------------------------------------------------------------------------------- /diffusion/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /diffusion/__pycache__/diffusion_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/__pycache__/diffusion_utils.cpython-39.pyc -------------------------------------------------------------------------------- /diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc -------------------------------------------------------------------------------- /diffusion/__pycache__/respace.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/__pycache__/respace.cpython-39.pyc -------------------------------------------------------------------------------- /diffusion/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/diffusion_utils.py -------------------------------------------------------------------------------- /diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/respace.py -------------------------------------------------------------------------------- /diffusion/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/diffusion/timestep_sampler.py -------------------------------------------------------------------------------- /download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/download.py -------------------------------------------------------------------------------- /evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/evaluator.py -------------------------------------------------------------------------------- /imgs/effect.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/imgs/effect.png -------------------------------------------------------------------------------- /imgs/scheme.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/imgs/scheme.png -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/models.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/sample_ddp.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/train.py -------------------------------------------------------------------------------- /train_accelerate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/train_accelerate.py -------------------------------------------------------------------------------- /udit_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/udit_models.py -------------------------------------------------------------------------------- /utils/fastdit_kit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/utils/fastdit_kit.py -------------------------------------------------------------------------------- /utils/parser_setter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/utils/parser_setter.py -------------------------------------------------------------------------------- /utils/test_latency_throughout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/utils/test_latency_throughout.py -------------------------------------------------------------------------------- /utils/udit_models_original.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YuchuanTian/U-DiT/HEAD/utils/udit_models_original.py --------------------------------------------------------------------------------