yangluo7 / CAME
The official implementation of "CAME: Confidence-guided Adaptive Memory Optimization"
☆84Updated 7 months ago
Alternatives and similar repositories for CAME:
Users that are interested in CAME are comparing it to the libraries listed below
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆92Updated 7 months ago
- ☆48Updated 11 months ago
- ☆144Updated last month
- LoRA fine-tune directly on the quantized models.☆26Updated 2 months ago
- Official codebase for Margin-aware Preference Optimization for Aligning Diffusion Models without Reference (MaPO).☆68Updated 8 months ago
- [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization☆32Updated 2 months ago
- ☆21Updated last month
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training☆174Updated 3 weeks ago
- Official implementation of "DoRA: Weight-Decomposed Low-Rank Adaptation"☆123Updated 9 months ago
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆35Updated 7 months ago
- TerDiT: Ternary Diffusion Models with Transformers☆68Updated 8 months ago
- TinyFusion: Diffusion Transformers Learned Shallow☆74Updated 2 months ago
- Low-bit optimizers for PyTorch☆125Updated last year
- Patch convolution to avoid large GPU memory usage of Conv2D☆85Updated 3 weeks ago
- A parallelism VAE avoids OOM for high resolution image generation☆53Updated 3 weeks ago
- QuEST: Efficient Finetuning for Low-bit Diffusion Models☆38Updated last month
- Code for NeurIPS 2023 paper "Restart Sampling for Improving Generative Processes"☆147Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆129Updated last year
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆27Updated 10 months ago
- ☆15Updated 2 months ago
- ☆74Updated 2 months ago
- The official implementation for "MonoFormer: One Transformer for Both Diffusion and Autoregression"☆84Updated 4 months ago
- PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu☆53Updated 2 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆58Updated 8 months ago
- Merge safetensor files using the technique described in "Language Models are Super Mario: Absorbing Abilities from Homologous Models as a…☆76Updated 4 months ago
- Accelerating Diffusion Transformers with Token-wise Feature Caching☆62Updated 2 weeks ago
- Adaptive Caching for Faster Video Generation with Diffusion Transformers☆142Updated 3 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆95Updated 4 months ago
- Triton implement of bi-directional (non-causal) linear attention☆41Updated 2 weeks ago