Anima-Lab / MaskDiT
Code for Fast Training of Diffusion Models with Masked Transformers
☆398Updated 11 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☆280Updated this week
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆342Updated 3 months ago
- [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy☆245Updated 4 months ago
- ☆280Updated 6 months ago
- Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)☆559Updated last year
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆220Updated 2 weeks ago
- Consistency Models Made Easy☆278Updated 6 months ago
- Scalable Diffusion Models with State Space Backbone☆152Updated last year
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆214Updated last month
- MoVQGAN - model for the image encoding and reconstruction☆233Updated last year
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"☆807Updated last year
- An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch☆300Updated 2 weeks ago
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆450Updated 2 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆372Updated 5 months ago
- Official Pytorch Implementation of Our CVPR2023 Paper: "Towards Accurate Image Coding: Improved Autoregressive Image Generation with Dyna…☆172Updated last year
- ☆457Updated 4 months ago
- ☆210Updated last year
- [CVPR'23] Video Probabilistic Diffusion Models in Projected Latent Space☆315Updated 11 months ago
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆115Updated last year
- Official pytorch implementation of the paper: "An Edit Friendly DDPM Noise Space: Inversion and Manipulations". CVPR 2024.☆332Updated 9 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆183Updated last month
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆939Updated last month
- PyTorch code and model checkpoints for Score identity Distillation (SiD) and its adversarial version (SiDA)☆113Updated 3 weeks ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆692Updated 4 months ago
- 🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"☆239Updated 2 years ago
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆337Updated 6 months ago
- ☆177Updated 2 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆94Updated last year
- [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model☆411Updated 5 months ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆266Updated 3 months ago