kvfrans / shortcut-models
☆170Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for shortcut-models
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆184Updated last week
- Transformer implementation for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆55Updated 3 weeks ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆182Updated last month
- Consistency Models Made Easy☆221Updated 3 weeks ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆150Updated 4 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆292Updated last week
- Rectified Diffusion: Straightness Is Not Your Need☆118Updated 2 weeks ago
- The codebase of our paper "Improving the Training of Rectified Flows"☆80Updated 3 weeks ago
- A mini-library for training consistency models.☆213Updated 10 months ago
- Scalable Diffusion Models with State Space Backbone☆145Updated 8 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆591Updated 3 weeks ago
- ☆108Updated last year
- Train VAE like a boss☆242Updated 3 weeks ago
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆346Updated last year
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆34Updated last month
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆253Updated 2 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆88Updated 9 months ago
- Annotated Flow Matching paper☆130Updated last month
- ☆231Updated last month
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆523Updated 5 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- Implementation of a framework for Gamengen in Pytorch☆90Updated last month
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆132Updated 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…☆241Updated last week
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆198Updated last week
- Code for "Diffusion Model Alignment Using Direct Preference Optimization"☆261Updated 10 months ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆38Updated 10 months ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆372Updated 5 months ago
- ☆212Updated last year