attention-map-diffusers

可视化 Stable Diffusion 注意力图，理解 AI 绘画的决策过程

这是一个面向 Hugging Face Diffusers 的注意力图（Attention Map）可视化与分析工具库。它解决了 Stable Diffusion 等扩散模型在生成图像时内部注意力机制不透明的问题，帮助开发者和研究者理解文本提示词与生成图像区域之间的对应关系。核心能力包括：提取跨注意力层（cross-attention）的注意力图、将注意力图与生成图像对齐并可视化、支持按文本 token 查看对应区域的注意力热力图，以及提供 API 供二次开发和集成。通过该工具，用户可以直观地看到模型在生成过程中关注了哪些像素，从而用于调试提示词、分析模型行为、研究生成可控性等场景。项目基于 Python 实现，深度集成 Diffusers 生态，适合 AIGC 领域的开发者和研究者使用。