Anima-Lab / MaskDiT
Code for Fast Training of Diffusion Models with Masked Transformers
☆396Updated 10 months ago
Alternatives and similar repositories for MaskDiT:
Users that are interested in MaskDiT are comparing it to the libraries listed below
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆263Updated last month
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆553Updated 11 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆328Updated 2 months ago
- Consistency Models Made Easy☆276Updated 5 months ago
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆790Updated last year
- ☆275Updated 5 months ago
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆201Updated 3 weeks ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆201Updated last week
- EDM2 and Autoguidance -- Official PyTorch implementation☆663Updated 3 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆882Updated last week
- Scalable Diffusion Models with State Space Backbone☆152Updated last year
- Implementation of Autoregressive Diffusion in Pytorch☆364Updated 4 months ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆397Updated last month
- ☆411Updated 3 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆243Updated 3 months ago
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆407Updated 4 months ago
- MoVQGAN - model for the image encoding and reconstruction☆223Updated last year
- [CVPR'23] Video Probabilistic Diffusion Models in Projected Latent Space☆314Updated 10 months ago
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆297Updated last year
- This repo contains the code for 1D tokenizer and generator☆748Updated last week
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆112Updated 11 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆93Updated last year
- ☆211Updated last year
- Implementation of MagViT2 Tokenizer in Pytorch☆597Updated 2 months ago
- [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation☆488Updated 4 months ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆278Updated 4 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆223Updated 3 months ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆259Updated 2 months ago
- Official Pytorch Implementation of Our CVPR2023 Paper: "Towards Accurate Image Coding: Improved Autoregressive Image Generation with Dyna…☆167Updated last year
- Scaling Diffusion Transformers with Mixture of Experts☆294Updated 6 months ago