Alternatives and similar repositories for shortcut-models:

Users that are interested in shortcut-models are comparing it to the libraries listed below

• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆908Updated last month
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆277Updated 2 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆803Updated 2 weeks ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆682Updated 4 months ago
• locuslab / ect
  Consistency Models Made Easy
  ☆278Updated 6 months ago
• sony / ctm
  ☆279Updated 6 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆369Updated 5 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆338Updated 3 months ago
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆429Updated 3 weeks ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆821Updated 3 weeks ago
• G-U-N / Rectified-Diffusion
  [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need
  ☆210Updated last month
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆800Updated last year
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆398Updated 11 months ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆674Updated last week
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆207Updated 3 months ago
• TencentARC / SEED-Voken
  SEED-Voken: A Series of Powerful Visual Tokenizers
  ☆863Updated last month
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,197Updated 8 months ago
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆308Updated 7 months ago
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆490Updated 5 months ago
• apple / ml-tarflow
  ☆161Updated 4 months ago
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,467Updated 6 months ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆242Updated last year
• lucidrains / magvit2-pytorch
  Implementation of MagViT2 Tokenizer in Pytorch
  ☆599Updated 3 months ago
• kwsong0113 / diffusion-forcing-transformer
  Official PyTorch Implementation of "History-Guided Video Diffusion"
  ☆260Updated last month
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆546Updated last year
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆341Updated 4 months ago
• Hhhhhhao / continuous_tokenizer
  ☆176Updated 2 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆557Updated 11 months ago
• jannerm / ddpo
  Code for the paper "Training Diffusion Models with Reinforcement Learning"
  ☆440Updated last year
• feizc / DiS
  Scalable Diffusion Models with State Space Backbone
  ☆152Updated last year