← all repositories

adobe-research/custom-diffusion

A method for fine-tuning text-to-image diffusion models with few images to add new concepts or styles.

custom-diffusion
Velocity · 7d
+1.5
★ / day
Trend
steady
star history

Custom Diffusion enables few-shot fine-tuning of Stable Diffusion and other text-to-image diffusion models given 4-20 images of a new concept. The method optimizes only key and value projection matrices in cross-attention layers, enabling fast training (~6 minutes on 2 A100 GPUs) and lightweight per-concept storage (75MB). The approach supports single and multi-concept customization including object-style combinations and new category additions.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.