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 Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆65Updated 3 years ago
- JAX implementation ViT-VQGAN☆82Updated 3 years ago
- Official Code Implementation for 'A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models'☆20Updated last year
- Implementation of NWT, audio-to-video generation, in Pytorch☆92Updated 3 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆135Updated 2 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆97Updated 3 years ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆122Updated 4 months ago
- Official PyTorch implementation for Maximum Likelihood Training of Implicit Nonlinear Diffusion Model (INDM) in NeurIPS 2022.☆40Updated last year
- ☆143Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆207Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆89Updated 10 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆107Updated last year
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆93Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆96Updated last year
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆160Updated 2 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
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆62Updated last year
- Implementation of the video diffusion model and training scheme presented in the paper, Flexible Diffusion Modeling of Long Videos, in Py…☆85Updated 3 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆49Updated last year
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆94Updated 6 months ago
- The official implementation of MAGVLT: Masked Generative Vision-and-Language Transformer (CVPR'23)☆28Updated last year
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆58Updated 2 years ago
- ☆42Updated last year
- [ICLR2022] Code for "Retriever: Learning Content-Style Representation as a Token-Level Bipartite Graph"☆54Updated 3 years ago
- Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)☆193Updated 3 years ago
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆50Updated last year
- Implementation of Agent Attention in Pytorch☆93Updated last year