ComfyUI_KV_Edit

KV_Edit: Training-Free Image Editing for Precise Background Preservation，you can use it in comfyUI

Update

• edit some code for comfy 修改一些代码，用起来更简单吧。注意，unet模型位置改变了

1. Installation

In the ./ComfyUI/custom_node directory, run the following:

git clone https://github.com/smthemex/ComfyUI_KV_Edit

---

2. Requirements

pip install -r requirements.txt

3.Model

• 3.1.1 download 'ae.safetensor' and 'flux1-dev.safetensors' from here 下载ae和flux单体模型模型,文件结构如下图
• 3.1.2 download 'clip_l.safetensors' and 't5xxl_fp8_e4m3fn.safetensors' from here下载T5和clip l单体模型模型,文件结构如下图

--   ComfyUI/models/diffusion_models  # or unet
    ├── flux1-dev.safetensors #23.8G 目前不能用量化版的单体，还需修改代码，全量的也能跑，慢点而已
--   ComfyUI/models/clip
    ├── clip_l.safetensors
    ├── t5xxl_fp8_e4m3fn.safetensorsvae
--   ComfyUI/models/vae
    ├── ae.safetensor

Example

• use comfy T5 and clip and flux dev single checkpoints用comfy的text encoder和flux的fp16模型，推荐
• use comfy or flux dev single checkpoints 仅使用单体模型，速度更慢，更占内存，主要是量化两次模型，目前fp8模式无法使用

Citation

If you find our work helpful, please star 🌟 this repo and cite 📑 our paper. Thanks for your support!

@article{zhu2025kv,
  title={KV-Edit: Training-Free Image Editing for Precise Background Preservation},
  author={Zhu, Tianrui and Zhang, Shiyi and Shao, Jiawei and Tang, Yansong},
  journal={arXiv preprint arXiv:2502.17363},
  year={2025}
}

👍🏻 Acknowledgements

Our code is modified based on FLUX and RF-Solver-Edit. Special thanks to Wenke Huang for his early inspiration and helpful guidance to this project!

📧 Contact

This repository is currently under active development and restructuring. The codebase is being optimized for better stability and reproducibility. While we strive to maintain code quality, you may encounter temporary issues during this transition period. For any questions or technical discussions, feel free to open an issue or contact us via email at [email protected].