├── .gitignore ├── LICENSE.txt ├── README.md ├── config └── 000_DiffMoE_S_E16_Flow.yaml ├── dataset └── dataset_cache.py ├── diffusion ├── __init__.py ├── diffusion_utils.py ├── gaussian_diffusion.py ├── rectified_flow.py ├── respace.py └── timestep_sampler.py ├── download.py ├── environment.yml ├── evaluation ├── evaluator.py ├── fid.py └── inception.py ├── figs ├── logo.png ├── method.png └── teaser.png ├── models ├── models_DiT.py ├── models_DiffMoE.py ├── models_ECDiT.py ├── models_TCDiT.py └── modules.py ├── sample_ddp_feature.py ├── train.py ├── train.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/README.md -------------------------------------------------------------------------------- /config/000_DiffMoE_S_E16_Flow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/config/000_DiffMoE_S_E16_Flow.yaml -------------------------------------------------------------------------------- /dataset/dataset_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/dataset/dataset_cache.py -------------------------------------------------------------------------------- /diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/__init__.py -------------------------------------------------------------------------------- /diffusion/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/diffusion_utils.py -------------------------------------------------------------------------------- /diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /diffusion/rectified_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/rectified_flow.py -------------------------------------------------------------------------------- /diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/respace.py -------------------------------------------------------------------------------- /diffusion/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/diffusion/timestep_sampler.py -------------------------------------------------------------------------------- /download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/download.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/evaluation/evaluator.py -------------------------------------------------------------------------------- /evaluation/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/evaluation/fid.py -------------------------------------------------------------------------------- /evaluation/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/evaluation/inception.py -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/figs/method.png -------------------------------------------------------------------------------- /figs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/figs/teaser.png -------------------------------------------------------------------------------- /models/models_DiT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/models/models_DiT.py -------------------------------------------------------------------------------- /models/models_DiffMoE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/models/models_DiffMoE.py -------------------------------------------------------------------------------- /models/models_ECDiT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/models/models_ECDiT.py -------------------------------------------------------------------------------- /models/models_TCDiT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/models/models_TCDiT.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/models/modules.py -------------------------------------------------------------------------------- /sample_ddp_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/sample_ddp_feature.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/train.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KlingTeam/DiffMoE/HEAD/utils.py --------------------------------------------------------------------------------