sayakpaul/kv-caching-flow-models-benchmarks
395 MB
Sayak Paul PRO
sayakpaul
AI & ML interests
Diffusion models, representation learning
Recent Activity
updated a collection about 11 hours ago
QAI HF Hackathon Doha Submissions updated a collection about 11 hours ago
QAI HF Hackathon Doha Submissions updated a collection about 11 hours ago
QAI HF Hackathon Doha SubmissionsOrganizations
Posts 24
Post
3,612
Fast LoRA inference for Flux with Diffusers and PEFT 🚨
There are great materials that demonstrate how to optimize inference for popular image generation models, such as Flux. However, very few cover how to serve LoRAs fast, despite LoRAs being an inseparable part of their adoption.
In our latest post, @BenjaminB and I show different techniques to optimize LoRA inference for the Flux family of models for image generation. Our recipe includes the use of:
1.
2. Flash Attention 3 (when compatible)
3. Dynamic FP8 weight quantization (when compatible)
4. Hotswapping for avoiding recompilation during swapping new LoRAs 🤯
We have tested our recipe with Flux.1-Dev on both H100 and RTX 4090. We achieve at least a *2x speedup* in either of the GPUs. We believe our recipe is grounded in the reality of how LoRA-based use cases are generally served. So, we hope this will be beneficial to the community 🤗
Even though our recipe was tested primarily with NVIDIA GPUs, it should also work with AMD GPUs.
Learn the details and the full code here:
https://huggingface.co/blog/lora-fast
There are great materials that demonstrate how to optimize inference for popular image generation models, such as Flux. However, very few cover how to serve LoRAs fast, despite LoRAs being an inseparable part of their adoption.
In our latest post, @BenjaminB and I show different techniques to optimize LoRA inference for the Flux family of models for image generation. Our recipe includes the use of:
1.
torch.compile2. Flash Attention 3 (when compatible)
3. Dynamic FP8 weight quantization (when compatible)
4. Hotswapping for avoiding recompilation during swapping new LoRAs 🤯
We have tested our recipe with Flux.1-Dev on both H100 and RTX 4090. We achieve at least a *2x speedup* in either of the GPUs. We believe our recipe is grounded in the reality of how LoRA-based use cases are generally served. So, we hope this will be beneficial to the community 🤗
Even though our recipe was tested primarily with NVIDIA GPUs, it should also work with AMD GPUs.
Learn the details and the full code here:
https://huggingface.co/blog/lora-fast
spaces 18
Running
Agents
1
Diffusers Project Pulse
🎨
Launch an interactive web dashboard for the app
Running
Batch image captioning on TPU
🔥
vLLM TPU, Qwen2.5-VL, and the Torchax fallback path
Sleeping
Stem Separator
🎵
Separate audio tracks into individual stems
Runtime error
Agents
Stem Separator Inference
🏆
Separate audio into individual stems (vocals, drums, etc.)
Running
NB Fashion Moodboard
💻
Nano Banana powered fashion moodboard
Running
Agents
6
Civitai To Hub
🐠
Upload checkpoints from CivitAI to Hugging Face Hub.
models 28
sayakpaul/nanoDiT-JAX-butterflies
Updated • 34
sayakpaul/FluxTransformer2DModel-sm120-cu130-r54
Updated
sayakpaul/different-lora-from-civitai
12B • Updated • 60 • 1
sayakpaul/flux-diffusers-gguf
12B • Updated • 60 • 1
sayakpaul/trained-lumina2-lora-yarn
Text-to-Image • Updated • 73 • 3
sayakpaul/FLUX.1-dev-edit-v0
Text-to-Image • 12B • Updated • 40 • 46
sayakpaul/cartoon-control-lr_1e-4-wd_1e-4-gs_10.0-cd_0.1
Text-to-Image • 12B • Updated • 35 • 6
sayakpaul/q8-ltx-video
2B • Updated • 22 • 7
sayakpaul/bnb-single-file-checkpoint-from-civitai
6B • Updated • 35
sayakpaul/mochi-lora-dissolve
Text-to-Video • Updated • 57 • 2
datasets 26
sayakpaul/sample-datasets
Viewer • Updated • 6 • 45.8k • 1
sayakpaul/ltx2-optims-artifacts
Viewer • Updated • 7 • 179
sayakpaul/torch-profiling-trace-diffusers
Viewer • Updated • 18 • 190
sayakpaul/diffusers-blackwell-quants
Updated • 547 • 1
sayakpaul/ShapeNet-55
Viewer • Updated • 52.5k • 17.2k
sayakpaul/llm-bias-mt-bench
Viewer • Updated • 4.76k • 62
sayakpaul/butteflies_with_classes
Preview • Updated • 2.56k
sayakpaul/OmniEdit-mini
Viewer • Updated • 21.1k • 195 • 6
sayakpaul/video-dataset-disney-organized
Viewer • Updated • 69 • 348 • 6
sayakpaul/pick-a-pic-v2-unique-prompts
Viewer • Updated • 59k • 177 • 2