cloneofsimo / vqgan-training
Train VAE like a boss
☆276Updated 6 months ago
Alternatives and similar repositories for vqgan-training:
Users that are interested in vqgan-training are comparing it to the libraries listed below
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆517Updated 10 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆349Updated 3 months ago
- Official implementation of Inductive Moment Matching☆458Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated last week
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆226Updated 5 months ago
- Focused on fast experimentation and simplicity☆71Updated 4 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆167Updated last month
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆137Updated 2 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆172Updated 10 months ago
- ☆44Updated 2 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆90Updated 3 months ago
- Text to Image Latent Diffusion using a Transformer core☆181Updated 8 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆401Updated 11 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆699Updated 4 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆126Updated last month
- WIP☆93Updated 8 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆94Updated last month
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆94Updated last year
- My take on Flow Matching☆52Updated 3 months ago
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆420Updated last year
- DDT: Decoupled Diffusion Transformer☆224Updated 3 weeks ago
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆125Updated last year
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆112Updated 3 months ago
- Official Implementation of weights2weights☆141Updated last month
- Inference-time scaling of diffusion-based image and video generation models.☆142Updated 2 months ago
- Consistency Models Made Easy☆277Updated 6 months ago
- ☆174Updated 4 months ago
- MoVQGAN - model for the image encoding and reconstruction☆233Updated last year
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆504Updated 11 months ago
- Scalable Diffusion Models with State Space Backbone☆152Updated last year