cloneofsimo / vqgan-training
Train VAE like a boss
☆270Updated 4 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☆489Updated 8 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆322Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆262Updated last month
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆222Updated 3 months ago
- Text to Image Latent Diffusion using a Transformer core☆171Updated 6 months ago
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆115Updated 11 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆168Updated 8 months ago
- Focused on fast experimentation and simplicity☆69Updated 2 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆396Updated 10 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆77Updated 2 months ago
- This is a PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Framework for Cross-Modality Evolu…☆139Updated 3 weeks ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆93Updated last year
- Official Implementation of weights2weights☆140Updated last week
- ☆43Updated 3 weeks ago
- AlignProp uses direct reward backpropogation for the alignment of large-scale text-to-image diffusion models. Our method is 25x more samp…☆276Updated 4 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆659Updated 3 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆104Updated last month
- ☆31Updated last year
- My take on Flow Matching☆41Updated 2 months ago
- Huggingface-compatible SDXL Unet implementation that is readily hackable☆412Updated last year
- Consistency Models Made Easy☆275Updated 5 months ago
- WIP☆93Updated 7 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 8 months ago
- Smooth Diffusion: Crafting Smooth Latent Spaces in Diffusion Models arXiv 2023 / CVPR 2024☆333Updated 5 months ago
- MoVQGAN - model for the image encoding and reconstruction☆223Updated last year