lucidrains / titok-pytorchLinks
Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
☆182Updated last year
Alternatives and similar repositories for titok-pytorch
Users that are interested in titok-pytorch are comparing it to the libraries listed below
Sorting:
- Implementation of a multimodal diffusion transformer in Pytorch☆105Updated last year
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 10 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆253Updated 4 months ago
- ☆137Updated last year
- ☆70Updated 10 months ago
- Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch☆54Updated 10 months ago
- DDT: Decoupled Diffusion Transformer☆276Updated last month
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆151Updated 8 months ago
- ☆181Updated 9 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated last year
- ☆128Updated last month
- Implementation of a framework for Genie2 in Pytorch☆153Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆452Updated 8 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆124Updated 6 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆170Updated 4 months ago
- Official PyTorch implementation of TokenSet.☆125Updated 6 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆123Updated 11 months ago
- Adaptive Length Image Tokenization via Recurrent Allocation | How many tokens is an image worth ?☆132Updated 7 months ago
- A Video Tokenizer Evaluation Dataset☆134Updated 8 months ago
- [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination☆102Updated 2 months ago
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆122Updated 4 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆140Updated 8 months ago
- [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Mo…☆303Updated 4 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆237Updated 5 months ago
- Towards training VQ-VAE models robustly!☆84Updated 2 months ago
- ☆151Updated 6 months ago
- The official implementation of "[MASK] is All You Need"☆124Updated 2 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆122Updated 5 months ago
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆61Updated 4 months ago
- (ICCV 2025) "Principal Components" Enable A New Language of Images☆63Updated 2 months ago