comfyui-daam

在 ComfyUI 中可视化每个词对生成图像的注意力影响

comfyui-daam 是一个 ComfyUI 的自定义节点插件，它将 Diffusion Attentive Attribution Maps (DAAM) 技术集成到 ComfyUI 工作流中。DAAM 是一种用于解释 Stable Diffusion 模型内部注意力机制的可视化方法，能够生成文本提示词与图像区域之间的注意力热力图。该插件解决了 ComfyUI 用户难以直观理解生成图像中每个词对应区域的问题，提供了注意力图的可视化节点，帮助用户调试提示词、分析模型行为。核心能力包括：在 ComfyUI 节点图中插入 DAAM 节点，实时输出注意力热力图，支持不同层和头的注意力聚合，并可与现有生成流程无缝衔接。项目基于 Python 开发，依赖 ComfyUI 和 Stable Diffusion 生态，适合研究模型可解释性和进行精细化图像生成的开发者。