ICEdit

:open_book: For more visual results, go checkout our project page

🎆 News

• [2025/9/19] 🔥 We have open-sourced our MoE version ICEdit and ckpt. Have a try!🚀
• [2025/9/18] 🌟 ICEdit has been accepted by NeurIPS 2025!🎉 See you in San Diego!
• [2025/8/21] 🌟 We have released an Ascend (Huawei NPU)-powered version of ICEdit. Now you can run ICEdit on Ascend NPU! Many thanks to Zhiyuan！
• [2025/5/16] 🌟 Many thanks to gluttony-10 (十字鱼) for adapting Gradio demo with GGUF quantization, further reducing memory usage to 10GB.
• [2025/5/14] 🔥 With the help of the official comfy-org, we have integrated our ComfyUI nodes into Comfy Registry!
• [2025/5/13] 🔥 We have released the training code! Train your own editing LoRAs now!
• [2025/5/11] 🌟 Great thanks to gluttony-10 (十字鱼) for making a windows gradio demo to use our project on Windows!
• [2025/5/8] 🔥 We have released our official ComfyUI workflow! 🚀 Check the repository and have a try!

• [2025/5/8] 🔥 We have added LoRA scale slider in the gradio demo. You can try to discover more interesting demo with different scale!

• [2025/5/7] 🌟 We update some notes when using the ComfyUI workflow to avoid unsatisfactory results!
• [2025/5/6] 🔥 ICEdit currently ranks 2nd on the overall/weekly trending list of Hugging Face space. Thank you all for your support and love!🤗
• [2025/5/5] 🌟 Heartfelt thanks to Datou for creating a fantastic ComfyUI workflow on OpenArt! 🚀 Have a try!
• [2025/5/2] 🌟 Heartfelt thanks to judian17 for crafting an amazing ComfyUI-nunchaku demo! Only 4GB VRAM GPU is enough to run with ComfyUI-nunchaku!🚀 Dive in and give it a spin!
• [2025/4/30] 🔥 We release the Huggingface Demo 🤗! Have a try!
• [2025/4/30] 🔥 We release the paper on arXiv!
• [2025/4/29] We release the project page and demo video! Codes will be made available in next week~ Happy Labor Day!

🎈 Tutorial on Bilibili or Youtube

👑 Feel free to share your results in this Gallery!

• [2025/5/15] 🌟 We find that 啦啦啦的小黄瓜 has made a detailed bilibili tutorial introducing our model! What a great video!
• [2025/5/14] 🌟 We find that Nenly同学 has made a fantastic bilibili tutorial on how to use our repository! Great thanks to him!
• [2025/5/10] 🌟 Great thanks to 月下Hugo for making a Chinese tutorial on how to use our official workflow!
• [2025/5/7] 🌟 Heartfelt thanks to T8star for making a tutorial and ComfyUI workflow on how to increase the editing success to 100%!🚀 Have a try!
• [2025/5/3] 🌟 Heartfelt thanks to softicelee2 for making a Youtube video on how to use our model!

📖 Table of Contents

• 🎆 News
• 🎈 Tutorial on Bilibili or Youtube
  • 👑 Feel free to share your results in this Gallery!
• 📖 Table of Contents
  • 📢 Attention All: Incorrect ComfyUI Workflow Usage Alert!
• 💼 Installation
  • Conda environment setup
  • Download pretrained weights
  • Inference in bash (w/o VLM Inference-time Scaling)
    • For the usage of MoE-LoRA version
  • Inference in Gradio Demo
  • 💼 Windows one-click package
• 🔧 Training
• 🎨ComfyUI Workflow
  • Official ComfyUI-workflow
  • ComfyUI-workflow for increased editing success rate
  • ComfyUI-nunchaku
  • ComfyUI-workflow
• ⚠️ Tips
  • If you encounter such a failure case, please try again with a different seed!
  • ⚠️ Clarification
• 💪 To Do List
• 💪 Comparison with Commercial Models
• 🌟 Star History
• Bibtex

📢 Attention All: Incorrect ComfyUI Workflow Usage Alert!

• We have released our official ComfyUI workflow for proper usage! Check our repository and have a try!

• You need to add the fixed pre-prompt "A diptych with two side-by-side images of the same scene. On the right, the scene is exactly the same as on the left but {instruction}" before inputing the edit instructions, otherwise you may get bad results! (This is mentioned in the paper!, The code for the Hugging Face gradio demo already embeds this prompt. So, you can simply input the editing instructions without additional setup.)
• The width of the input image must resize to 512 (no restriction to height).
• Please use the Normal LoRA not the MoE-LoRA, because the MoE-LoRA cannot be correctly loaded with ComfyUI lora loader.
• 🔥💐🎆 Welcome to share your creative workflows (such as combining Redux, ACE, etc.) in the Issues section and showcase the results! We will include references so that more people can see your creativity.

💼 Installation

Conda environment setup

conda create -n icedit python=3.10
conda activate icedit
pip install -r requirements.txt
pip install -U huggingface_hub

Download pretrained weights

If you can connect to Huggingface, you don't need to download the weights. Otherwise, you need to download the weights to local.

• Flux.1-fill-dev.
• ICEdit-normal-LoRA.
• ICEdit-MoE-LoRA

Note: Due to some cooperation permission issues, we have to withdraw the weights and codes of moe-lora temporarily. What is released currently is just the ordinary lora, but it still has powerful performance. If you urgently need the moe lora weights of the original text, please email the author.

Inference in bash (w/o VLM Inference-time Scaling)

Now you can have a try!

Our model can only edit images with a width of 512 pixels (there is no restriction on the height). If you pass in an image with a width other than 512 pixels, the model will automatically resize it to 512 pixels.

If you found the model failed to generate the expected results, please try to change the --seed parameter. Inference-time Scaling with VLM can help much to improve the results.

python scripts/inference.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --seed 304897401 \

Editing a 512×768 image requires 35 GB of GPU memory. If you need to run on a system with 24 GB of GPU memory (for example, an NVIDIA RTX3090), you can add the --enable-model-cpu-offload parameter.

python scripts/inference.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --enable-model-cpu-offload

If you have downloaded the pretrained weights locally, please pass the parameters during inference, as in:

python scripts/inference.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --flux-path /path/to/flux.1-fill-dev \
                            --lora-path /path/to/ICEdit-normal-LoRA

For the usage of MoE-LoRA version

python scripts/inference_moe.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --seed 42 \

python scripts/inference_moe.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --enable-model-cpu-offload

python scripts/inference_moe.py --image assets/girl.png \
                            --instruction "Make her hair dark green and her clothes checked." \
                            --flux-path /path/to/flux.1-fill-dev \
                            --lora-path /path/to/ICEdit-MoE-LoRA

Inference in Gradio Demo

We provide a gradio demo for you to edit images in a more user-friendly way. You can run the following command to start the demo.

python scripts/gradio_demo.py --port 7860



## for MoE version
python scripts/gradio_demo_moe.py --port 7860

Like the inference script, if you want to run the demo on a system with 24 GB of GPU memory, you can add the --enable-model-cpu-offload parameter. And if you have downloaded the pretrained weights locally, please pass the parameters during inference, as in:

python scripts/gradio_demo.py --port 7860 \
                              --flux-path /path/to/flux.1-fill-dev (optional) \
                              --lora-path /path/to/ICEdit-normal-LoRA (optional) \
                              --enable-model-cpu-offload (optional) \

## for MoE version
python scripts/gradio_demo_moe.py --port 7860 \
                              --flux-path /path/to/flux.1-fill-dev (optional) \
                              --lora-path /path/to/ICEdit-normal-LoRA (optional) \
                              --enable-model-cpu-offload (optional) \

Or if you want to run the demo on a system with 10 GB of GPU memory, you can download the gguf models from FLUX.1-Fill-dev-gguf, t5-v1_1-xxl-encoder-gguf and pass the parameters during inference, as in:

python scripts/gradio_demo.py --port 7861 \
                              --flux-path models/flux.1-fill-dev \
                              --lora-path models/ICEdit-normal-LoRA \
                              --transformer models/flux1-fill-dev-Q4_0.gguf \
                              --text_encoder_2 models/t5-v1_1-xxl-encoder-Q8_0.gguf \
                              --enable-model-cpu-offload \

Then you can open the link in your browser to edit images.

Here is also a Chinese tutorial Youtube video on how to install and use ICEdit, created by softicelee2. It's definitely worth a watch!

💼 Windows one-click package

Great thanks to gluttony-10, a famous Bilibili Up! He made a tutorial (Youtube and Bilibili) on how to install our project on windows and a one-click package for Windows! Just unzip it and it's ready to use. It has undergone quantization processing. It only takes up 14GB of space and supports graphics cards of the 50 series.

Download link: Google Drive or Baidu Wangpan(refer to the comment section of the video)

🔧 Training

Found more details in here: Training Code

🎨ComfyUI Workflow

Official ComfyUI-workflow

We have released our official ComfyUI workflow in this repository for correct usage of our model! We have embedded the prompt "A diptych with two side-by-side images of the same scene ... but" into our nodes and you just need to input the edit instructions such as "make the girl wear pink sunglasses". We also add a high resolution refinement module for better image quality! The total VRAM consumption is about 14GB. Use this workflow and the ICEdit-normal-lora to fulfill your creative ideas!

We have specially created a repository for the workflow and you can install it directly in ComfyUI. Just open the manager tab and click 'Install via Git URL', copy the following URL and you are able to use it. For more details please refer to this issue

URL: https://github.com/hayd-zju/ICEdit-ComfyUI-official

Great thanks to 月下Hugo for making a Chinese tutorial on how to use our official workflow!

ComfyUI-workflow for increased editing success rate

Thanks to T8star! He made a tutorial (Youtube and bilibili) and a creative workflow (OpenArt and RunningHub) that could increase the editing success rate greatly (about 100%)! Have a try with it!

ComfyUI-nunchaku

We extend our heartfelt thanks to @judian17 for crafting a ComfyUI workflow that facilitates seamless usage of our model. Explore this excellent workflow to effortlessly run our model within ComfyUI. Only 4GB VRAM GPU is enough to run with ComfyUI-nunchaku!

This workflow incorporates high-definition refinement, yielding remarkably good results. Moreover, integrating this LoRA with Redux enables outfit changes to a certain degree. Once again, a huge thank you to @judian17 for his innovative contributions!

ComfyUI-workflow

Thanks to Datou, a workflow of ICEdit in ComfyUI can also be downloaded here. Try it with the normal lora ckpt.

⚠️ Tips

If you encounter such a failure case, please try again with a different seed!

• Our base model, FLUX, does not inherently support a wide range of styles, so a large portion of our dataset involves style transfer. As a result, the model may sometimes inexplicably change your artistic style.

• Our training dataset is mostly targeted at realistic images. For non-realistic images, such as anime or blurry pictures, the success rate of the editing drop and could potentially affect the final image quality.

• While the success rates for adding objects, modifying color attributes, applying style transfer, and changing backgrounds are high, the success rate for object removal is relatively lower due to the low quality of the removal dataset we use.

The current model is the one used in the experiments in the paper, trained with only 4 A800 GPUs (total batch_size = 2 x 2 x 4 = 16). In the future, we will enhance the dataset, and do scale-up, finally release a more powerful model.

⚠️ Clarification

We've noticed numerous web pages related to ICEdit, including https://icedit.net/, https://icedit.org/. Kudos to those who built these pages!

However, we'd like to emphasize two important points:

• No Commercial Use: Our project cannot be used for commercial purposes. Please check the LICENSE for details.
• Official Page: The official project page is https://river-zhang.github.io/ICEdit-gh-pages/.

💪 To Do List

• Inference Code
• Inference-time Scaling with VLM
• Pretrained Weights
• More Inference Demos
• Gradio demo
• Comfy UI demo (by @judian17, compatible with nunchaku, support high-res refinement and FLUX Redux. Only 4GB VRAM GPU is enough to run!)
• Comfy UI demo with normal lora (by @Datou in openart)
• Official ComfyUI workflow
• Training Code
• LoRA for higher image resolution (768, 1024)

💪 Comparison with Commercial Models

🌟 Star History

Bibtex

If this work is helpful for your research, please consider citing the following BibTeX entry.

@article{zhang2025context,
  title={In-context edit: Enabling instructional image editing with in-context generation in large scale diffusion transformer},
  author={Zhang, Zechuan and Xie, Ji and Lu, Yu and Yang, Zongxin and Yang, Yi},
  journal={arXiv preprint arXiv:2504.20690},
  year={2025}
}

@inproceedings{zhang2025icedit,
  title     = {In-Context Edit: Enabling Instructional Image Editing with In-Context Generation in Large-Scale Diffusion Transformers},
  author    = {Zhang, Zechuan and Xie, Ji and Lu, Yu and Yang, Zongxin and Yang, Yi},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year      = {2025},
  note      = {arXiv:2504.20690}
}