lucidrains / RQ-TransformerLinks
Implementation of RQ Transformer, proposed in the paper "Autoregressive Image Generation using Residual Quantization"
☆122Updated 3 years ago
Alternatives and similar repositories for RQ-Transformer
Users that are interested in RQ-Transformer are comparing it to the libraries listed below
Sorting:
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆89Updated last year
- An implementation of simple diffusion in PyTorch (and JAX)☆35Updated 2 years ago
- Implementation of NWT, audio-to-video generation, in Pytorch☆92Updated 3 years ago
- JAX implementation ViT-VQGAN☆82Updated 3 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆134Updated 3 months ago
- Official Code Implementation for 'A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models'☆20Updated last year
- Official PyTorch implementation for Maximum Likelihood Training of Implicit Nonlinear Diffusion Model (INDM) in NeurIPS 2022.☆40Updated last year
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆65Updated 3 years ago
- Implementation of a multimodal diffusion transformer in Pytorch☆107Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆58Updated 2 years ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆89Updated 11 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆98Updated 4 years ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆207Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)☆193Updated 3 years ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆160Updated 2 years ago
- The official implementation of MAGVLT: Masked Generative Vision-and-Language Transformer (CVPR'23)☆28Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆98Updated last year
- Implementation of a Light Recurrent Unit in Pytorch☆49Updated last year
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆51Updated last year
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆64Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- A repo for "Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation"☆50Updated 3 years ago
- [ICCV 2023] Online Clustered Codebook☆181Updated last year
- ☆144Updated last year
- PyTorch implementation of slicing adversarial network (SAN)☆98Updated last month
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆125Updated 4 months ago
- ☆44Updated last year
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆93Updated 2 years ago