← all repositories

castorini/daam

A cross-attention visualization technique for interpreting Stable Diffusion image generation.

798 stars Jupyter Notebook Image · Video · Audio
daam
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

DAAM (Diffusion Attentive Attribution Maps) generates attention-based heatmaps that reveal how Stable Diffusion models respond to different parts of a text prompt when generating images. The tool hooks into the cross-attention layers of diffusion models to produce attribution maps that highlight which word tokens most influence different regions of the generated output. It supports the standard Stable Diffusion models as well as SDXL, and provides both a Python library for integration and a CLI/demo interface.

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