cloneofsimo / vqgan-training
Train VAE like a boss
☆274Updated 5 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☆508Updated 9 months ago
- Official implementation of Inductive Moment Matching☆436Updated last month
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆337Updated 3 months ago
- Text to Image Latent Diffusion using a Transformer core☆176Updated 7 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆277Updated 2 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆227Updated 4 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆161Updated 3 weeks ago
- Focused on fast experimentation and simplicity☆71Updated 3 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆398Updated 10 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆170Updated 9 months ago
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆119Updated last year
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆123Updated last month
- The aim of this repository is to test and implement Flow-Matching-based models☆85Updated 2 months ago
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆416Updated last year
- WIP☆93Updated 8 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆110Updated 2 months ago
- ☆44Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆682Updated 4 months ago
- T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!☆393Updated last month
- Consistency Models Made Easy☆277Updated 6 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆94Updated last year
- ☆19Updated 5 months ago
- Official implementation of "Controlling Text-to-Image Diffusion by Orthogonal Finetuning".☆289Updated 5 months ago
- Official Implementation of weights2weights☆140Updated last month
- ☆51Updated last year
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆80Updated last week
- A mini-library for training consistency models.☆241Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆203Updated last year
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆336Updated 6 months ago
- My take on Flow Matching☆50Updated 3 months ago