sayakpaul/dit-xl2-samples
11 GB
Sayak Paul PRO
sayakpaul
AI & ML interests
Diffusion models, representation learning
Recent Activity
liked a model about 5 hours ago
Qwen/Qwen-Image-2.1 liked a Space about 11 hours ago
imstevenpmwork/lerobot-adoption-dashboard updated a dataset 3 days ago
huggingface/diffusers-metadataOrganizations
Posts 24
Post
3436
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
Provides a list of papers focusing on optimizing T2I diffusion models, targeting fewer timesteps, architecture optimization, and more.
-
Progressive Distillation for Fast Sampling of Diffusion Models
Paper • 2202.00512 • Published • 2 -
On Distillation of Guided Diffusion Models
Paper • 2210.03142 • Published -
InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generation
Paper • 2309.06380 • Published • 34 -
Consistency Models
Paper • 2303.01469 • Published • 9
Contains Flux Fill, Canny, and Dev checkpoints in NF4.
Provides a list of papers focusing on optimizing T2I diffusion models, targeting fewer timesteps, architecture optimization, and more.
-
Progressive Distillation for Fast Sampling of Diffusion Models
Paper • 2202.00512 • Published • 2 -
On Distillation of Guided Diffusion Models
Paper • 2210.03142 • Published -
InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generation
Paper • 2309.06380 • Published • 34 -
Consistency Models
Paper • 2303.01469 • Published • 9
spaces 17
Running
Batch image captioning on TPU
🔥
vLLM TPU, Qwen2.5-VL, and the Torchax fallback path
Running
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.
Sleeping
Agents
9
Grade Images with Gemini
🚀
Uses Gemini 2.0 Flash to grade images.
models 27
sayakpaul/FluxTransformer2DModel-sm120-cu130-r54
Updated
sayakpaul/different-lora-from-civitai
12B • Updated • 26 • 1
sayakpaul/flux-diffusers-gguf
12B • Updated • 54 • 1
sayakpaul/trained-lumina2-lora-yarn
Text-to-Image • Updated • 24 • 3
sayakpaul/FLUX.1-dev-edit-v0
Text-to-Image • 12B • Updated • 32 • 46
sayakpaul/cartoon-control-lr_1e-4-wd_1e-4-gs_10.0-cd_0.1
Text-to-Image • 12B • Updated • 54 • 6
sayakpaul/q8-ltx-video
2B • Updated • 36 • 7
sayakpaul/bnb-single-file-checkpoint-from-civitai
6B • Updated • 38
sayakpaul/mochi-lora-dissolve
Text-to-Video • Updated • 24 • 2
sayakpaul/flux.1-dev-int8-aot-compiled
Updated • 4
datasets 26
sayakpaul/sample-datasets
Viewer • Updated • 6 • 11.3k • 1
sayakpaul/ltx2-optims-artifacts
Viewer • Updated • 7 • 154
sayakpaul/torch-profiling-trace-diffusers
Viewer • Updated • 18 • 244
sayakpaul/diffusers-blackwell-quants
Updated • 527 • 1
sayakpaul/ShapeNet-55
Viewer • Updated • 52.5k • 869
sayakpaul/llm-bias-mt-bench
Viewer • Updated • 4.76k • 106
sayakpaul/butteflies_with_classes
Preview • Updated • 1.62k
sayakpaul/OmniEdit-mini
Viewer • Updated • 21.1k • 581 • 6
sayakpaul/video-dataset-disney-organized
Viewer • Updated • 69 • 516 • 6
sayakpaul/pick-a-pic-v2-unique-prompts
Viewer • Updated • 59k • 307 • 2