Name		Name	Last commit message	Last commit date
parent directory ..
diffusion		diffusion
DiT_README.md		DiT_README.md
README.md		README.md
download.py		download.py
dpm_sample.py		dpm_sample.py
environment.yml		environment.yml
eval.sh		eval.sh
evaluation.py		evaluation.py
models.py		models.py
sample_ddp.py		sample_ddp.py
train.py		train.py
train.sh		train.sh

README.md

Guidance Free Training script for DiT

Environment / Dataset Setup

Exactly the same as DiT. See DiT_README.md

Model Evaluation

Pretrained models

model	reso.	FID (w/o CFG)	HF weights🤗
DiT-B/2 (pretrain)	256	9.04	DiT-B-GF-scratch.pt
DiT-XL/2 (finetune)	256	1.99	DiT-XL-GF-finetune.pt

Evaluation

Please refer to DiT/eval.sh for usage reference.

Model Training

Please refer to DiT/train.sh for usage reference.

Acknowledgements

This codebase is partially adapted from the DiT. We thank the authors for their clean and well-documented implementation.