explainingai-code / DiT-PyTorch
This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
☆29Updated 4 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:
- Train VAE like a boss☆278Updated 6 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆321Updated 8 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆528Updated 10 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆351Updated 4 months ago
- An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL☆526Updated this week
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆286Updated 3 weeks ago
- Text to Image Latent Diffusion using a Transformer core☆183Updated 8 months ago
- This repo implements Video generation model using Latent Diffusion Transformers(Latte) in PyTorch and provides training and inference cod…☆12Updated 4 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated 10 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆76Updated 6 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆377Updated 6 months ago
- ☆182Updated 5 months ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆581Updated 7 months ago
- Implementation of the paper "Denoising Diffusion Probabilistic Models" in PyTorch☆60Updated last year
- Focused on fast experimentation and simplicity☆72Updated 4 months ago
- DDT: Decoupled Diffusion Transformer☆237Updated last month
- The aim of this repository is to test and implement Flow-Matching-based models☆92Updated 4 months ago
- ☆492Updated 5 months ago
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆56Updated last month
- ☆122Updated 10 months ago
- ☆283Updated 7 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆221Updated 2 months ago
- Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers☆194Updated last month
- MoVQGAN - model for the image encoding and reconstruction☆233Updated last year
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆136Updated 4 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆402Updated last year
- PyTorch Implementation of Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model☆21Updated 7 months ago
- Official implementation of Inductive Moment Matching☆465Updated 2 months ago
- Minimal Implementation of a D3PM in pytorch☆222Updated last year
- Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆595Updated last month