lucidrains / RQ-Transformer
Implementation of RQ Transformer, proposed in the paper "Autoregressive Image Generation using Residual Quantization"
☆99Updated 2 years ago
Alternatives and similar repositories for RQ-Transformer:
Users that are interested in RQ-Transformer are comparing it to the libraries listed below
- JAX implementation ViT-VQGAN☆80Updated 2 years ago
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆82Updated 3 months ago
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆64Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- An implementation of simple diffusion in PyTorch (and JAX)☆35Updated last year
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆61Updated last week
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆92Updated 3 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆98Updated last month
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆27Updated 10 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆177Updated this week
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆34Updated 9 months ago
- Jax/Flax implementation of Variational-DiffWave.☆40Updated 2 years ago
- Implementation of a multimodal diffusion transformer in Pytorch☆99Updated 6 months ago
- ☆51Updated last year
- The official implementation of MAGVLT: Masked Generative Vision-and-Language Transformer (CVPR'23)☆26Updated 11 months ago
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆55Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆197Updated 11 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆77Updated 7 months ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆93Updated 11 months ago
- A repo for "Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation"☆50Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Codebase for the Paper: Learning Visual Styles from Audio-Visual Associations (ECCV 2022, in PyTorch)☆14Updated last year
- ☆44Updated 8 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆231Updated this week
- Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)☆55Updated 2 years ago
- ☆37Updated 8 months ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆155Updated last year
- ☆35Updated last year