IronMan's starred repositories
VAR
[GPT beats diffusionš„] [scaling laws in visual generationš] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-simple, user-friendly yet state-of-the-art* codebase for autoregressive image generation!
fastcomposer
FastComposer: Tuning-Free Multi-Subject Image Generation with Localized Attention
PAIR-Diffusion
[CVPR 2024] PAIR Diffusion: A Comprehensive Multimodal Object-Level Image Editor
Prompt-Diffusion
Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"
Perturbed-Attention-Guidance
Official implementation of "Perturbed-Attention Guidance"
moe_attention
Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"
attention-interpolation-diffusion
Interpolation Between Text-to-Image Generation!
PlainMamba
PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition
DiffusionNAG
Official PyTorch implementation of "DiffusionNAG: Predictor-guided Neural Architecture Generation with Diffusion Models" (ICLR 2024)
optin-transformer-pruning
[ICLR 2024] The Need for Speed: Pruning Transformers with One Recipe
RepresentationSurgery
Representation Surgery for Multi-Task Model Merging. ICML, 2024.