kvfrans / shortcut-models
☆195Updated last month
Related projects ⓘ
Alternatives and complementary repositories for shortcut-models
- Transformer implementation for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆55Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆191Updated 2 weeks ago
- Consistency Models Made Easy☆226Updated last month
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆185Updated last month
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆615Updated last week
- Rectified Diffusion: Straightness Is Not Your Need☆128Updated this week
- A mini-library for training consistency models.☆215Updated 10 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆300Updated 2 weeks ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆255Updated 2 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆151Updated 4 months ago
- Train VAE like a boss☆246Updated 3 weeks ago
- The codebase of our paper "Improving the Training of Rectified Flows"☆82Updated last month
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆532Updated 5 months ago
- Code for the paper "Training Diffusion Models with Reinforcement Learning"☆354Updated last year
- Scalable Diffusion Models with State Space Backbone☆146Updated 8 months ago
- ☆238Updated last month
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆35Updated last month
- Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆664Updated this week
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆88Updated 9 months ago
- Implementation of a framework for Gamengen in Pytorch☆90Updated 2 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- Annotated Flow Matching paper☆135Updated 2 months ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆434Updated 7 months ago
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆99Updated 7 months ago
- Minimal Implementation of a D3PM in pytorch☆183Updated 6 months ago
- ☆110Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆161Updated 5 months ago
- A summary of related works about flow matching, stochastic interpolants☆337Updated 3 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆110Updated 2 months ago
- ☆211Updated last year