lucidrains / titok-pytorchLinks
Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
☆176Updated 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☆102Updated last year
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 8 months ago
- Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch☆52Updated 8 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆234Updated 2 months ago
- ☆70Updated 8 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆129Updated 6 months ago
- Implementation of a framework for Genie2 in Pytorch☆149Updated 6 months ago
- ☆132Updated last year
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆401Updated 6 months ago
- ☆137Updated 4 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆119Updated 9 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆165Updated 2 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆139Updated 6 months ago
- [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination☆90Updated this week
- Towards training VQ-VAE models robustly!☆79Updated 3 weeks ago
- Adaptive Length Image Tokenization via Recurrent Allocation | How many tokens is an image worth ?☆126Updated 5 months ago
- [ICLR 2025] Source code for paper "A Spark of Vision-Language Intelligence: 2-Dimensional Autoregressive Transformer for Efficient Finegr…☆76Updated 7 months ago
- Scalable Diffusion Models with State Space Backbone☆156Updated last year
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆112Updated 4 months ago
- [NeurIPS 2024] Official PyTorch Implementation of "FlowTurbo: Towards Real-time Flow-Based Image Generation with Velocity Refiner"☆71Updated 10 months ago
- A Video Tokenizer Evaluation Dataset☆129Updated 6 months ago
- ☆177Updated 7 months ago
- DDT: Decoupled Diffusion Transformer☆268Updated last month
- (ICCV 2025) "Principal Components" Enable A New Language of Images☆54Updated last week
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆113Updated 2 months ago
- Official PyTorch Implementation for Paper "No More Adam: Learning Rate Scaling at Initialization is All You Need"☆52Updated 6 months ago
- Scaling RWKV-Like Architectures for Diffusion Models☆136Updated last year
- Code for the paper "Vamba: Understanding Hour-Long Videos with Hybrid Mamba-Transformers" [ICCV 2025]☆78Updated last week
- Autoregressive Image Generation with Randomized Parallel Decoding☆70Updated 4 months ago