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取出来