lucidrains / RQ-Transformer
Implementation of RQ Transformer, proposed in the paper "Autoregressive Image Generation using Residual Quantization"
☆94Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for RQ-Transformer
- An implementation of simple diffusion in PyTorch (and JAX)☆35Updated last year
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆81Updated 3 weeks ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 2 years ago
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆64Updated 2 years ago
- Implementation of NWT, audio-to-video generation, in Pytorch☆87Updated 2 years ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆150Updated 4 months ago
- Official PyTorch implementation for FastDPM, a fast sampling algorithm for diffusion probabilistic models☆81Updated 3 years ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆26Updated 8 months ago
- [ICLR2022] Code for "Retriever: Learning Content-Style Representation as a Token-Level Bipartite Graph"☆53Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- Jax/Flax implementation of Variational-DiffWave.☆40Updated 2 years ago
- Locally Hierarchical Auto-Regressive Modeling for Image Generation (HQ-Transformer)☆26Updated 8 months ago
- [NeurIPS 2023] Official Implementation: "Consistent Diffusion Models"☆54Updated last year
- Unofficial implementation of Neural Analysis and Synthesis☆7Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- [ICCV 2021] A Pytorch implementation of "Manifold Matching via Deep Metric Learning for Generative Modeling"☆79Updated last year
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- A convolution-free, transformer-only version of the CycleGAN framework☆32Updated 2 years ago
- ☆50Updated 10 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆69Updated 3 months ago
- ☆29Updated 11 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆94Updated 2 weeks ago
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆68Updated last year
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆76Updated 5 months ago
- A repo for "Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation"☆51Updated 2 years ago
- ☆32Updated 9 months ago
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆27Updated 6 months ago