Sign in

Nodes Browser

ComfyDeploy: How ComfyUI_VisualAttentionMap works in ComfyUI?

What is ComfyUI_VisualAttentionMap?

NODES:HF ModelLoader, Show Images, Text2Image Inference, Decode Latent, Show CrossAttn Map, Show SelfAttn Map

How to install it in ComfyDeploy?

Head over to the machine page

  1. Click on the "Create a new machine" button
  2. Select the Edit build steps
  3. Add a new step -> Custom Node
  4. Search for ComfyUI_VisualAttentionMap and select it
  5. Close the build step dialig and then click on the "Save" button to rebuild the machine

ComfyUI nodes to use VisualAttentionMap

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

image

Install

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

  • 手动安装：

    1. cd custom_nodes
    2. git clone https://github.com/leeguandong/ComfyUI_VisualAttentionMap.git
    3. 重启 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

Star History Chart