explainingai-code / DiT-PyTorchLinks
This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
☆39Updated 5 months ago
Alternatives and similar repositories for DiT-PyTorch
Users that are interested in DiT-PyTorch are comparing it to the libraries listed below
Sorting:
- Text to Image Latent Diffusion using a Transformer core☆191Updated 10 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆314Updated last week
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆592Updated 11 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆340Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆370Updated 5 months ago
- Train VAE like a boss☆281Updated 8 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆229Updated 3 months ago
- Official implementation of Inductive Moment Matching☆488Updated 3 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆272Updated 2 months ago
- DDT: Decoupled Diffusion Transformer☆258Updated 2 weeks ago
- The aim of this repository is to test and implement Flow-Matching-based models☆96Updated 5 months ago
- My Implementation of Adversarial Diffusion Distillation https://arxiv.org/pdf/2311.17042.pdf☆73Updated 6 months ago
- This repo implements a Stable Diffusion model in PyTorch with all the essential components.☆214Updated 7 months ago
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆234Updated 6 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆520Updated last year
- An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆783Updated last week
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆270Updated 2 weeks ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆603Updated 8 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆392Updated 7 months ago
- This is a repo to track the latest autoregressive visual generation papers.☆356Updated this week
- [Preprint] UCGM: Unified Continuous Generative Models☆156Updated last month
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆107Updated 2 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆137Updated 5 months ago
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆65Updated 3 months ago
- Implementation of a framework for Genie2 in Pytorch☆150Updated 5 months ago
- Official Implementation for Diffusion Models Without Classifier-free Guidance☆130Updated 4 months ago
- ☆553Updated 6 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆215Updated last month
- (CVPR 2025) Switti: Designing Scale-Wise Transformers for Text-to-Image Synthesis☆185Updated 3 months ago
- Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch☆277Updated 11 months ago