Nikolai10 / FSQ
TensorFlow implementation of "Finite Scalar Quantization: VQ-VAE Made Simple" (ICLR 2024)
☆15Updated last year
Alternatives and similar repositories for FSQ:
Users that are interested in FSQ are comparing it to the libraries listed below
- A Pytorch Implementation of Finite Scalar Quantization☆104Updated last year
- The official PyTorch implementation of Fast Diffusion Model☆94Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆92Updated 3 months ago
- "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆45Updated 2 months ago
- [ICCV 2023] Online Clustered Codebook☆157Updated 4 months ago
- ☆112Updated 7 months ago
- Accelerating Diffusion Models via Early Stop of the Diffusion Process☆36Updated 2 years ago
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆35Updated 9 months ago
- Official Pytorch Implementation of Our CVPR2023 Paper: "Not All Image Regions Matter: Masked Vector Quantization for Autoregressive Image…☆57Updated last year
- Inference-only implementation of "One-Step Diffusion Distillation through Score Implicit Matching" [NIPS 2024]☆77Updated 2 months ago
- ☆43Updated 4 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆77Updated 8 months ago
- Implementation of the paper "MaskBit: Embedding-free Image Generation from Bit Tokens"☆41Updated last month
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆55Updated last year
- official code for Diff-Instruct algorithm for one-step diffusion distillation☆67Updated 3 weeks ago
- code based for rectified flow☆32Updated this week
- [ICLR 2024] Official code for the paper 'Elucidating the Exposure Bias in Diffusion Models'☆40Updated 8 months ago
- ☆44Updated last month
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆41Updated 3 months ago
- “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…☆84Updated last month
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆27Updated 10 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated 7 months ago
- Official Repository of IJCAI 2024 Paper: "BATON: Aligning Text-to-Audio Model with Human Preference Feedback"☆18Updated 3 weeks ago
- [ICLR 2024] Official implementation of Bellman Optimal Stepsize Straightening of Flow-Matching Models☆34Updated 11 months ago
- The official codebase for Reflected Flow Matching (ICML 2024)☆13Updated 7 months ago
- Official code for "On Calibrating Diffusion Probabilistic Models"☆28Updated last year
- Towards training VQ-VAE models robustly!☆43Updated 3 weeks ago
- ☆133Updated last month
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆107Updated 10 months ago
- [ICML 2024] Improving Diffusion Models for Inverse Problems Using Optimal Posterior Covariance☆44Updated 3 months ago