cloneofsimo / vqgan-training
Train VAE like a boss
☆242Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for vqgan-training
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆443Updated 4 months ago
- Text to Image Latent Diffusion using a Transformer core☆141Updated 2 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆182Updated last month
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆253Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆181Updated last week
- ☆166Updated 3 weeks ago
- Repo is required for the code of our research paper on micro-budget training of large scale diffusion model.☆153Updated 3 months ago
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆523Updated 5 months ago
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆97Updated 7 months ago
- WIP☆89Updated 2 months ago
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆398Updated last year
- Minimal Implementation of a D3PM in pytorch☆181Updated 6 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆370Updated 5 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆161Updated 4 months ago
- PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)☆442Updated 5 months ago
- Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch☆246Updated 3 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆88Updated 9 months ago
- HART: Efficient Visual Generation with Hybrid Autoregressive Transformer☆324Updated 3 weeks ago
- Consistency Models Made Easy☆220Updated 3 weeks ago
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆317Updated last month
- Official Implementation of weights2weights☆121Updated last month
- A mini-library for training consistency models.☆213Updated 10 months ago
- A suite of image and video neural tokenizers☆622Updated this week
- Scaling Diffusion Transformers with Mixture of Experts☆202Updated 2 months ago
- T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!☆359Updated 2 months ago
- ☆26Updated 6 months ago
- Unofficial Implementation of Consistency Models in pytorch☆250Updated last year
- Simple large-scale training of stable diffusion with multi-node support.☆125Updated last year
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆241Updated last week
- ☆31Updated 11 months ago