explainingai-code / DiT-PyTorch
This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
☆24Updated last month
Alternatives and similar repositories for DiT-PyTorch:
Users that are interested in DiT-PyTorch are comparing it to the libraries listed below
- Text to Image Latent Diffusion using a Transformer core☆162Updated 5 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆75Updated 3 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆307Updated last month
- Implementation of the paper "Denoising Diffusion Probabilistic Models" in PyTorch☆49Updated last year
- Focused on fast experimentation and simplicity☆66Updated last month
- Implementation of a multimodal diffusion transformer in Pytorch☆100Updated 7 months ago
- ☆43Updated last month
- Scaling Diffusion Transformers with Mixture of Experts☆260Updated 5 months ago
- This repo implements a Stable Diffusion model in PyTorch with all the essential components.☆183Updated 2 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆196Updated 3 weeks ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆247Updated last week
- Train VAE like a boss☆265Updated 4 months ago
- Minimal Implementation of a D3PM in pytorch☆196Updated 9 months ago
- From scratch implementation of a vision language model in pure PyTorch☆194Updated 9 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆73Updated this week
- The code and models for the paper: Switti: Designing Scale-Wise Transformers for Text-to-Image Synthesis☆160Updated last month
- Video+code lecture on building nanoGPT from scratch☆65Updated 8 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆356Updated 3 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆73Updated last month
- Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch☆156Updated last month
- ☆24Updated 2 months ago
- PyTorch Implementation of Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model☆17Updated 4 months ago
- Transformer-Mamba Diffusion Models☆98Updated 7 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆124Updated last month
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆478Updated 7 months ago
- Explorations into improving ViTArc with Slot Attention☆37Updated 4 months ago
- Repository of lessons exploring image diffusion models, focused on understanding and education.☆44Updated last month
- When it comes to optimizers, it's always better to be safe than sorry☆179Updated 3 weeks ago
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆94Updated last year
- Implementation of Diffusion Transformer Model in Pytorch☆52Updated 2 months ago