ComfyUI_VisualAttentionMap

ComfyUI_VisualAttentionMap
★ 8

注意力可视化Prompt-to-PromptComfyUI节点交叉/自注意力
在ComfyUI中生成并展示Cross/Self注意力热图，基于Prompt-to-Prompt方法帮助分析与编辑提示词。
💡 在文本到图像流程中实时查看和调试注意力热图。
🍴 1 Forks💻 Python🔄 2024-08-26
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/a1f1f564f19c
image
Star History Chart
📄 README

ComfyUI nodes to use VisualAttentionMap

https://github.com/google/prompt-to-prompt

Install

  • 推荐使用管理器 ComfyUI Manager 安装（On the Way）

  • 手动安装：
  • cd custom_nodes
  • git clone https://github.com/leeguandong/ComfyUI_VisualAttentionMap.git
  • 重启 ComfyUI

    • Use

  • 节点：
  • HF ModelLoader: 加载hf格式的模型权重
  • Text2Image Inference: 推理，并保存了attention map
  • Decode Latent: VAE解码
  • Show Images: 展示attention map
  • Show CrossAttn Map: 把cross attention map取出来
  • Show SelfAttn Map: 把self attention map取出来

    • Stars

    [](https://star-history.com/#leeguandong/ComfyUI_VisualAttentionMap&Date)