explainingai-code / DiT-PyTorch
This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
☆20Updated 2 weeks ago
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☆158Updated 4 months ago
- Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch☆262Updated 5 months ago
- Focused on fast experimentation and simplicity☆64Updated 3 weeks ago
- Implementation of the paper "Denoising Diffusion Probabilistic Models" in PyTorch☆45Updated last year
- Implementation of a multimodal diffusion transformer in Pytorch☆99Updated 6 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆71Updated 2 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆62Updated 2 weeks ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆286Updated last week
- Scaling Diffusion Transformers with Mixture of Experts☆243Updated 4 months ago
- Train VAE like a boss☆252Updated 2 months ago
- This repo implements a Stable Diffusion model in PyTorch with all the essential components.☆164Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆233Updated last week
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆92Updated last year
- Minimal Implementation of a D3PM in pytorch☆192Updated 8 months ago
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆186Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆465Updated 6 months ago
- LoRA and DoRA from Scratch Implementations☆194Updated 10 months ago
- ☆95Updated 2 weeks ago
- From scratch implementation of a vision language model in pure PyTorch☆189Updated 8 months ago
- PyTorch implementation of CLIP Maximum Mean Discrepancy (CMMD) for evaluating image generation models.☆112Updated 9 months ago
- ☆40Updated this week
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆189Updated 2 weeks ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆119Updated 5 months ago
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆181Updated 2 months ago
- This repo implements Denoising Diffusion Probabilistic Models (DDPM) in Pytorch☆92Updated last month
- Code for Fast Training of Diffusion Models with Masked Transformers☆385Updated 8 months ago
- From-scratch diffusion model implemented in PyTorch.☆81Updated 11 months ago
- ☆24Updated last month
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆98Updated last month
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆92Updated 4 months ago