MegEngine / MegDiffusion
MegEngine implementation of Diffusion Models.
☆16Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for MegDiffusion
- # Unified Normalization (ACM MM'22) By Qiming Yang, Kai Zhang, Chaoxiang Lan, Zhi Yang, Zheyang Li, Wenming Tan, Jun Xiao, and Shiliang P…☆34Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- An object detection codebase based on MegEngine.☆28Updated last year
- Keras implement of Finite Scalar Quantization☆63Updated last year
- Patch convolution to avoid large GPU memory usage of Conv2D☆79Updated 5 months ago
- TVMScript kernel for deformable attention☆24Updated 2 years ago
- useful dotfiles included vim, zsh, tmux and vscode☆17Updated 3 weeks ago
- ☆35Updated 5 months ago
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆75Updated 4 months ago
- Inference-only implementation of "One-Step Diffusion Distillation through Score Implicit Matching" [NIPS 2024]☆36Updated this week
- Page for the CVPR 2023 Tutorial - Efficient Neural Networks: From Algorithm Design to Practical Mobile Deployments☆13Updated last year
- Slides with modifications for a course at Tsinghua University.☆57Updated 2 years ago
- ☆24Updated last year
- ☆41Updated 8 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆50Updated 5 months ago
- A parallelism VAE avoids OOM for high resolution image generation☆40Updated last month
- A codebase & model zoo for pretrained backbone based on MegEngine.☆32Updated last year
- NeRF implementation in MegEngine☆9Updated 2 years ago
- Open(MM)DiT: An Easy, Fast and Memory-Efficient System for (MM)DiT Training and Inference☆21Updated 8 months ago
- A light-weight and high-efficient training framework for accelerating diffusion tasks.☆41Updated 2 months ago
- A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.☆13Updated last year
- Training LLaMA language model with MMEngine! It supports LoRA fine-tuning!☆40Updated last year
- Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)☆125Updated last year
- ☆100Updated 2 months ago
- Document the demo and a series of documents for learning the diffusion model.☆39Updated last year
- [ICLR 2022] "As-ViT: Auto-scaling Vision Transformers without Training" by Wuyang Chen, Wei Huang, Xianzhi Du, Xiaodan Song, Zhangyang Wa…☆76Updated 2 years ago
- Unofficial implementation for the paper 'Improving Diffusion Models for Inverse Problems using Manifold Constraints'[https://arxiv.org/ab…☆12Updated 2 years ago
- A Deep Learning Project about cats.☆11Updated 2 years ago
- ☆19Updated 3 years ago
- minisora-DiT, a DiT reproduction based on XTuner from the open source community MiniSora☆38Updated 7 months ago