milmor/diffusion-transformer
Implementation of Diffusion Transformer Model in Pytorch
Replaces standard attention with Linformer to reduce computational complexity, enabling training on single-GPU setups. Implements DiT blocks with adaptive layer normalization (adaLN-Zero) and uses the EDM sampler for improved diffusion scheduling. Includes FID evaluation metrics for image generation quality assessment on ImageNet-64, with plans for classifier-free guidance and latent diffusion variants.
Stars
73
Forks
11
Language
Python
License
MIT
Category
Last pushed
Jan 06, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/milmor/diffusion-transformer"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
quantgirluk/aleatory
📦 Python library for Stochastic Processes Simulation and Visualisation
TuftsBCB/RegDiffusion
Diffusion model for gene regulatory network inference.
blei-lab/treeffuser
Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression...
yuanchenyang/smalldiffusion
Simple and readable code for training and sampling from diffusion models
NVlabs/FastGen
NVIDIA FastGen: Fast Generation from Diffusion Models