patil-suraj / vit-vqgan
JAX implementation ViT-VQGAN
☆82Updated 2 years ago
Alternatives and similar repositories for vit-vqgan:
Users that are interested in vit-vqgan are comparing it to the libraries listed below
- CLOOB training (JAX) and inference (JAX and PyTorch)☆71Updated 2 years ago
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorch☆64Updated 2 years ago
- Implementation of the video diffusion model and training scheme presented in the paper, Flexible Diffusion Modeling of Long Videos, in Py…☆84Updated 2 years ago
- ☆28Updated 3 years ago
- Un-*** 50 billions multimodality dataset☆24Updated 2 years ago
- An implementation of simple diffusion in PyTorch (and JAX)☆35Updated 2 years ago
- FID computation in Jax/Flax.☆27Updated 8 months ago
- ☆51Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- Pedagogical codebase for a simplified score-based generative model design, with training loop☆39Updated 3 years ago
- Simple python template☆41Updated 11 months ago
- ☆157Updated 2 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Training simple models to predict CLIP image embeddings from text embeddings, and vice versa.☆60Updated 3 years ago
- Inverts CLIP text embeds to image embeds and visualizes with deep-image-prior.☆35Updated 2 years ago
- PyTorch implementation of Contrastive Feature Loss for Image Prediction (AIM Workshop at ICCV 2021)☆53Updated 3 years ago
- ☆31Updated last year
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.☆72Updated 3 years ago
- Finetune the 1.4B latent diffusion text2img-large checkpoint from CompVis using deepspeed. (work-in-progress)☆36Updated 2 years ago
- A small repository demonstrating the use of Webdataset and Imagenet☆15Updated last year
- ☆36Updated 11 months ago
- Simple large-scale training of stable diffusion with multi-node support.☆131Updated last year
- The implementation for Accelerating Guided Diffusion Sampling with Splitting Numerical Methods (2023)☆48Updated 2 years ago
- 🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch☆50Updated 2 years ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆156Updated 2 years ago
- ☆29Updated 2 years ago
- ☆53Updated 2 years ago
- Contrastive Language-Image Pretraining☆142Updated 2 years ago