GitHub - dreaming-huang/DIT_CM: a consistency model implement base on DiT

This project combine DiT and Consistency Models

DiT samples 50steps DiT samples 4steps DiTCM samples 4steps

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
__pycache__		__pycache__
diffusion		diffusion
performance		performance
train_options		train_options
visuals		visuals
.gitignore		.gitignore
=2.0		=2.0
CODE_OF_CONDUCT.md		CODE_OF_CONDUCT.md
CONTRIBUTING.md		CONTRIBUTING.md
LICENSE.txt		LICENSE.txt
README.md		README.md
aa.py		aa.py
bb.py		bb.py
ditCM.py		ditCM.py
ditCM_fast.py		ditCM_fast.py
download.py		download.py
dreamgaussian_colab.ipynb		dreamgaussian_colab.ipynb
environment.yml		environment.yml
evaluator.py		evaluator.py
extract_features.py		extract_features.py
models.py		models.py
run_DiT.ipynb		run_DiT.ipynb
sample.png		sample.png
sample.py		sample.py
sample_ddp.py		sample_ddp.py
sample_nocfg.png		sample_nocfg.png
testCM.ipynb		testCM.ipynb
testCM.py		testCM.py
train.py		train.py
打印函数.py		打印函数.py