maproute-comfyui

MapRoute ComfyUI Node

MapRoute (CVPR 2026) 概念擦除方法的 ComfyUI 自定义节点。在 ComfyUI 工作流中加载训练好的 Mapper 权重，从生成结果中擦除指定概念。复现代码。

安装

cd ComfyUI/custom_nodes/
git clone https://github.com/[your-username]/ComfyUI-MapRoute.git MapRoute

使用

1. 准备权重文件

将训练好的 Mapper 权重（.pth 文件）放入 ComfyUI/models/maproute/ 目录下。

2. 在工作流中使用

Mapper 节点应置于 Checkpoint Loader 与提示词 CLIP 文本编码器之间。

3. A/B 对比验证效果

搭建两条文生图工作流对比有/无 Mapper 的生成结果：

Checkpoint Loader
├── MapRoute Load Mapper → CLIP Text Encode → KSampler → Save
└──                       CLIP Text Encode → KSampler → Save

同 seed、同 prompt、同采样设置，唯一区别是 Mapper 是否启用。

已知限制

• 仅支持 SD1.x 系列 CLIP 结构（SD1ClipModel）的部分模型
• 未测试过是否兼容 SDXL、Flux 等使用双 CLIP 或 T5 编码器的架构
• 由于 Router 部分暂未实现，每个节点仅支持加载一个 Mapper

引用

@inproceedings{li2026maproute,
  title={MapRoute: Semantic Routing for Precise Concept Erasure with Mapper},
  author={Li, et al.},
  booktitle={CVPR},
  year={2026}
}