Nikolai10 / FSQ
TensorFlow implementation of "Finite Scalar Quantization: VQ-VAE Made Simple" (ICLR 2024)
☆17Updated last year
Alternatives and similar repositories for FSQ:
Users that are interested in FSQ are comparing it to the libraries listed below
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆36Updated last year
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆49Updated 6 months ago
- Official code for Accelerating Diffusion Sampling with Optimized Time Steps (CVPR 2024)☆30Updated last year
- official code for Diff-Instruct algorithm for one-step diffusion distillation☆73Updated 4 months ago
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆47Updated last month
- Shortcut flow matching Pytorch implementation☆38Updated 4 months ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆32Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆110Updated 6 months ago
- Official Pytorch Implementation of Our CVPR2023 Paper: "Not All Image Regions Matter: Masked Vector Quantization for Autoregressive Image…☆57Updated last year
- ☆122Updated 10 months ago
- [ICLR 2024] Official code for the paper 'Elucidating the Exposure Bias in Diffusion Models'☆43Updated last month
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆84Updated last week
- The official PyTorch implementation of Fast Diffusion Model☆95Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆56Updated 2 years ago
- Towards training VQ-VAE models robustly!☆72Updated 4 months ago
- The official codebase for Reflected Flow Matching (ICML 2024)☆18Updated 10 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆35Updated 10 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆76Updated 5 months ago
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆35Updated last year
- Accelerating Diffusion Models via Early Stop of the Diffusion Process☆37Updated 2 years ago
- A Pytorch Implementation of Finite Scalar Quantization☆129Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆82Updated 2 months ago
- An official pytorch implementation of EACL2024 short paper "Flow Matching for Conditional Text Generation in a Few Sampling Steps"☆16Updated 11 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆112Updated 3 months ago
- ☆52Updated 7 months ago
- Official Code Implementation for 'A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models'☆18Updated 9 months ago
- MaskFlow: Discrete Flows For Flexible and Efficient Long Video Generation☆21Updated 2 months ago
- "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆49Updated 5 months ago
- ACDiT: Interpolating Autoregressive Conditional Modeling and Diffusion Transformer☆33Updated 4 months ago
- ☆22Updated last year