GitHub - stepfun-ai/Step1X-Edit: A SOTA open-source image editing model, which aims to provide comparable performance against the closed-source models like GPT-4o and Gemini 2 Flash.

🔥🔥🔥 News!!

Nov 26, 2025: 👋 We release Step1X-Edit-v1p2 (referred to as ReasonEdit-S in the paper), a native reasoning edit model with better performance on KRIS-Bench and GEdit-Bench. Technical report can be found here.

Models	GEdit-Bench			Kris-Bench
Models	G_SC⬆️	G_PQ⬆️	G_O⬆️	FK⬆️	CK⬆️	PK⬆️	Overall⬆️
Flux-Kontext-dev	7.16	7.37	6.51	53.28	50.36	42.53	49.54
Qwen-Image-Edit-2509	8.00	7.86	7.56	61.47	56.79	47.07	56.15
Step1X-Edit v1.1	7.66	7.35	6.97	53.05	54.34	44.66	51.59
Step1x-edit-v1p2-preview	8.14	7.55	7.42	60.49	58.81	41.77	52.51
Step1x-edit-v1p2 (base)	7.77	7.65	7.24	58.23	60.55	46.21	56.33
Step1x-edit-v1p2 (thinking)	8.02	7.64	7.36	59.79	62.76	49.78	58.64
Step1x-edit-v1p2 (thinking + reflection)	8.18	7.85	7.58	62.44	65.72	50.42	60.93

Sep 08, 2025: 👋 We release step1x-edit-v1p2-preview, a new version of Step1X-Edit with reasoning edit ability and better performance (report to be released soon), featuring:

Native Reasoning Edit Model: Combines instruction reasoning with reflective correction to handle complex edits more accurately. Performance on KRIS-Bench:

Models	Factual Knowledge ⬆️	Conceptual Knowledge ⬆️	Procedural Knowledge ⬆️	Overall ⬆️
Step1X-Edit v1.1	53.05	54.34	44.66	51.59
Step1x-edit-v1p2-preview	60.49	58.81	41.77	52.51
Step1x-edit-v1p2-preview (thinking)	62.24	62.25	44.43	55.21
Step1x-edit-v1p2-preview (thinking + reflection)	62.94	61.82	44.08	55.64

Improved image editing quality and better instruction-following performance. Performance on GEdit-Bench:

Models	G_SC ⬆️	G_PQ ⬆️	G_O ⬆️	Q_SC ⬆️	Q_PQ ⬆️	Q_O ⬆️
Step1X-Edit (v1.0)	7.13	7.00	6.44	7.39	7.28	7.07
Step1X-Edit (v1.1)	7.66	7.35	6.97	7.65	7.41	7.35
Step1x-edit-v1p2-preview	8.14	7.55	7.42	7.90	7.34	7.40

Jul 09, 2025: 👋 We’ve updated the step1x-edit model and released it as step1x-edit-v1p1 (diffusers version see here), featuring:
- Added support for text-to-image (T2I) generation tasks
- Improved image editing quality and better instruction-following performance. Quantitative evaluation on GEdit-Bench-EN (Full set). G_SC, G_PQ, and G_O refer to the metrics evaluated by GPT-4.1, while Q_SC, Q_PQ, and Q_O refer to the metrics evaluated by Qwen2.5-VL-72B. To facilitate reproducibility, we have released the intermediate results of our model evaluations.
  
  Models G_SC ⬆️ G_PQ ⬆️ G_O ⬆️ Q_SC ⬆️ Q_PQ ⬆️ Q_O ⬆️
  
  Step1X-Edit (v1.0) 7.13 7.00 6.44 7.39 7.28 7.07
  
  Step1X-Edit (v1.1) 7.66 7.35 6.97 7.65 7.41 7.35
Jun 17, 2025: 👋 Support for Teacache and parallel inference has been added.
May 22, 2025: 👋 Step1X-Edit now supports Lora finetuning on a single 24GB GPU now! A hand-fixing Lora for anime characters has also been released. Download Lora
Apr 30, 2025: 🎉 Step1X-Edit ComfyUI Plugin is available now, thanks for the community contribution! quank123wip/ComfyUI-Step1X-Edit & raykindle/ComfyUI_Step1X-Edit.
Apr 27, 2025: 🎉 With community support, we update the inference code and model weights of Step1X-Edit-FP8. meimeilook/Step1X-Edit-FP8 & rkfg/Step1X-Edit-FP8.
Apr 26, 2025: 🎉 Step1X-Edit is now live — you can try editing images directly in the online demo! Online Demo
Apr 25, 2025: 👋 We release the evaluation code and benchmark data of Step1X-Edit. Download GEdit-Bench
Apr 25, 2025: 👋 We release the inference code and model weights of Step1X-Edit. ModelScope & HuggingFace models.
Apr 25, 2025: 👋 We have made our technical report available as open source. Read

📖 Introduction

We introduce a state-of-the-art image editing model, Step1X-Edit, which aims to provide comparable performance against the closed-source models like GPT-4o and Gemini2 Flash. More specifically, we adopt the Multimodal LLM to process the reference image and user's editing instruction. A latent embedding has been extracted and integrated with a diffusion image decoder to obtain the target image. To train the model, we build a data generation pipeline to produce a high-quality dataset. For evaluation, we develop the GEdit-Bench, a novel benchmark rooted in real-world user instructions. Experimental results on GEdit-Bench demonstrate that Step1X-Edit outperforms existing open-source baselines by a substantial margin and approaches the performance of leading proprietary models, thereby making significant contributions to the field of image editing. More details please refer to our technical report.

Step1X-Edit: a unified image editing model performs impressively on various genuine user instructions.

⚡️ Quick Start

Make sure your transformers==4.55.0 (we tested on this version)
Install the diffusers package locally, according model version you want to use

Step1X-Edit-v1p2 (v1.2)

Install the diffusers package from the following command:

git clone -b step1xedit_v1p2 https://github.com/Peyton-Chen/diffusers.git
cd diffusers
pip install -e .

Here is an example for using the Step1X-Edit-v1p2 model to edit images:

import torch
from diffusers import Step1XEditPipelineV1P2
from diffusers.utils import load_image
pipe = Step1XEditPipelineV1P2.from_pretrained("stepfun-ai/Step1X-Edit-v1p2", torch_dtype=torch.bfloat16)
pipe.to("cuda")
print("=== processing image ===")
image = load_image("examples/0000.jpg").convert("RGB")
prompt = "add a ruby pendant on the girl's neck."
enable_thinking_mode=True
enable_reflection_mode=True
pipe_output = pipe(
    image=image,
    prompt=prompt,
    num_inference_steps=50,
    true_cfg_scale=6,
    generator=torch.Generator().manual_seed(42),
    enable_thinking_mode=enable_thinking_mode,
    enable_reflection_mode=enable_reflection_mode,
)
if enable_thinking_mode:
    print("Reformat Prompt:", pipe_output.reformat_prompt)
for image_idx in range(len(pipe_output.images)):
    pipe_output.images[image_idx].save(f"0001-{image_idx}.jpg", lossless=True)
    if enable_reflection_mode:
        print(pipe_output.think_info[image_idx])
        print(pipe_output.best_info[image_idx])
pipe_output.final_images[0].save(f"0001-final.jpg", lossless=True)

The results looks like:

Step1X-Edit-v1p2-preview (v1.2-preview)

Install the diffusers package from the following command:

git clone -b dev/MergeV1-2 https://github.com/Peyton-Chen/diffusers.git
cd diffusers
pip install -e .

Here is an example for using the Step1X-Edit-v1p2-preview model to edit images:

import torch
from diffusers import Step1XEditPipelineV1P2
from diffusers.utils import load_image
pipe = Step1XEditPipelineV1P2.from_pretrained("stepfun-ai/Step1X-Edit-v1p2-preview", torch_dtype=torch.bfloat16)
pipe.to("cuda")
print("=== processing image ===")
image = load_image("examples/0000.jpg").convert("RGB")
prompt = "add a ruby pendant on the girl's neck."
enable_thinking_mode=True
enable_reflection_mode=True
pipe_output = pipe(
    image=image,
    prompt=prompt,
    num_inference_steps=28,
    true_cfg_scale=4,
    generator=torch.Generator().manual_seed(42),
    enable_thinking_mode=enable_thinking_mode,
    enable_reflection_mode=enable_reflection_mode,
)
if enable_thinking_mode:
    print("Reformat Prompt:", pipe_output.reformat_prompt)
for image_idx in range(len(pipe_output.images)):
    pipe_output.images[image_idx].save(f"0001-{image_idx}.jpg", lossless=True)
    if enable_reflection_mode:
        print(pipe_output.think_info[image_idx])

Step1X-Edit-v1p1 (v1.1)

Install the diffusers package from the following command:

git clone -b step1xedit https://github.com/Peyton-Chen/diffusers.git
cd diffusers
pip install -e .

Here is an example for using the Step1X-Edit-v1p1 model to edit images:

import torch
from diffusers import Step1XEditPipeline
from diffusers.utils import load_image


pipe = Step1XEditPipeline.from_pretrained("stepfun-ai/Step1X-Edit-v1p1-diffusers", torch_dtype=torch.bfloat16)
pipe.to("cuda")

print("=== processing image ===")
image = load_image("examples/0000.jpg").convert("RGB")
prompt = "给这个女生的脖子上戴一个带有红宝石的吊坠。"
image = pipe(
    image=image,
    prompt=prompt,
    num_inference_steps=28,
    size_level=1024,
    guidance_scale=6.0,
    generator=torch.Generator().manual_seed(42),
).images[0]
image.save("0000.jpg")

The results will look like:

🌟 Advanced Usage

We use the original Step1X-Edit model as an example to demonstrate some advanced uses of the model. Other versions of the model may have different inference processes.

A1. Requirements

We test our model using torch==2.3.1 and torch==2.5.1 with cuda-12.1. Install requirements:

pip install -r requirements.txt

Install flash-attn, here we provide a script to help find the pre-built wheel suitable for your system.

python scripts/get_flash_attn.py

The script will generate a wheel name like flash_attn-2.7.2.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl, which could be found in the release page of flash-attn.

Then you can download the corresponding pre-built wheel and install it following the instructions in flash-attn.

A2. Reduce GPU Memory Usage

You can use the following scripts to edit images with reduced GPU memory usage.

bash scripts/run_examples.sh

The default script runs the inference code with non-quantified weights. If you want to save the GPU memory usage, you can 1) set the --quantized flag in the script, which will quantify the weights to fp8, or 2) set the --offload flag in the script to offload some modules to CPU.

The following table shows the GPU Memory Usage and speed for running Step1X-Edit model (batch size = 1, with cfg) with different configurations:

Model	Peak GPU Memory (512 / 786 / 1024)	28 steps w flash-attn(512 / 786 / 1024)
Step1X-Edit	42.5GB / 46.5GB / 49.8GB	5s / 11s / 22s
Step1X-Edit (FP8)	31GB / 31.5GB / 34GB	6.8s / 13.5s / 25s
Step1X-Edit (offload)	25.9GB / 27.3GB / 29.1GB	49.6s / 54.1s / 63.2s
Step1X-Edit (FP8 + offload)	18GB / 18GB / 18GB	35s / 40s / 51s

The model is tested on one H800 GPU.
We recommend to use GPUs with 80GB of memory for better generation quality and efficiency.

A3. Multi-GPU inference

For multi-GPU inference, you can use the following script:

bash scripts/run_examples_parallel.sh

You can change the number of GPUs (GPU), the configuration of xDiT (--ulysses_degree or --ring_degree or --cfg_degree), and whether to enable TeaCache acceleration (--teacache) in the script. The table below presents the speedup of several efficient methods on the Step1X-Edit model.

Model	Peak GPU Memory	28 steps
Step1X-Edit + TeaCache	49.6GB	16.78s
Step1X-Edit + xDiT (GPU=2)	50.2GB	12.81s
Step1X-Edit + xDiT (GPU=4)	52.9GB	8.17s
Step1X-Edit + TeaCache + xDiT (GPU=2)	50.7GB	8.94s
Step1X-Edit + TeaCache + xDiT (GPU=4)	54.2GB	5.82s

The model was tested on H800 series GPUs with a resolution of 1024.
TeaCache's default threshold of 0.2 provides a good balance between efficiency and performance.
xDiT employs both CFG Parallelism and Ring Attention when using 4 GPUs, but only utilizes CFG Parallelism when operating with 2 GPUs.

This default script runs the inference code on example inputs. The results will look like:

A4. Finetuning

Lora training script

Here is the the GPU memory cost during training with lora rank as 64 and batchsize as 1:

Precision of DiT	bf16 (512 / 786 / 1024)	fp8 (512 / 786 / 1024)
GPU Memory	29.7GB / 31.6GB / 33.8GB	19.8GB / 21.3GB / 23.6GB

The script ./scripts/finetuning.sh shows how to fine-tune the Step1X-Edit model. With our default strategy, it is possible to fine-tune Step1X-Edit with 1024 resolution on a single 24GB GPU. Our fine-tuning script is adapted from kohya-ss/sd-scripts.

bash ./scripts/finetuning.sh

The custom dataset is organized by ./library/data_configs/step1x_edit.toml. Here metadata_file contains all the training sampels, including the absolute paths of source images, absolute paths of target images and instructions.

The metadata_file should be a json file containing a dict as follows:

{
  <target image path, str>: {
    'ref_image_path': <source image path, str>
    'caption': <the editing instruction, str>
  }, 
  ...
}

Inference with Lora

To inference with Lora, simply add --lora <path to your lora weights> when using inference.py. For example:

python inference.py --input_dir ./examples \
    --model_path /data/work_dir/step1x-edit/ \
    --json_path ./examples/prompt_cn.json \
    --output_dir ./output_cn \
    --seed 1234 --size_level 1024 \
    --lora 20250521_001-lora256-alpha128-fix-hand-per-epoch/step1x-edit_test.safetensors

Here is an example for our pretrained Lora weights, which is designed for fixing corrupted hands of anime characters.

To reproduce the cases above, you can run the following scripts:

bash scripts/run_examples_fix_hand.sh

📊 Benchmark

We release GEdit-Bench as a new benchmark, grounded in real-world usages is developed to support more authentic and comprehensive evaluation. This benchmark, which is carefully curated to reflect actual user editing needs and a wide range of editing scenarios, enables more authentic and comprehensive evaluations of image editing models. The evaluation process and related code can be found in GEdit-Bench/EVAL.md. Part results of the benchmark are shown below:

🧩 Community Contributions

If you develop/use Step1X-Edit in your projects, welcome to let us know 🎉.

A detailed introduction blog of Step1X-Edit: Step1X-Edit执行流程 by liwenju0
FP8 model weights: meimeilook/Step1X-Edit-FP8 by meimeilook; rkfg/Step1X-Edit-FP8 by rkfg
Step1X-Edit ComfyUI Plugin: quank123wip/ComfyUI-Step1X-Edit by quank123wip; raykindle/ComfyUI_Step1X-Edit by raykindle
Training scripts: hobart07/Step1X-Edit_train by hobart07

📚 Citation

If you find the Step1X-Edit series helpful for your research or applications, please consider ⭐ starring the repository and citing our paper.

@article{yin2025reasonedit,
  title={ReasonEdit: Towards Reasoning-Enhanced Image Editing Models}, 
  author={Fukun Yin, Shiyu Liu, Yucheng Han, Zhibo Wang, Peng Xing, Rui Wang, Wei Cheng, Yingming Wang, Aojie Li, Zixin Yin, Pengtao Chen, Xiangyu Zhang, Daxin Jiang, Xianfang Zeng, Gang Yu},
  journal={arXiv preprint arXiv:2511.22625},
  year={2025}
}

@article{wu2025kris,
  title={KRIS-Bench: Benchmarking Next-Level Intelligent Image Editing Models},
  author={Wu, Yongliang and Li, Zonghui and Hu, Xinting and Ye, Xinyu and Zeng, Xianfang and Yu, Gang and Zhu, Wenbo and Schiele, Bernt and Yang, Ming-Hsuan and Yang, Xu},
  journal={arXiv preprint arXiv:2505.16707},
  year={2025}
}

@article{liu2025step1x-edit,
  title={Step1X-Edit: A Practical Framework for General Image Editing}, 
  author={Shiyu Liu and Yucheng Han and Peng Xing and Fukun Yin and Rui Wang and Wei Cheng and Jiaqi Liao and Yingming Wang and Honghao Fu and Chunrui Han and Guopeng Li and Yuang Peng and Quan Sun and Jingwei Wu and Yan Cai and Zheng Ge and Ranchen Ming and Lei Xia and Xianfang Zeng and Yibo Zhu and Binxing Jiao and Xiangyu Zhang and Gang Yu and Daxin Jiang},
  journal={arXiv preprint arXiv:2504.17761},
  year={2025}
}

Acknowledgement

We would like to express our sincere thanks to the contributors of Kohya, SD3, FLUX, Qwen, xDiT, TeaCache, diffusers and HuggingFace teams, for their open research and exploration.

Disclaimer

The results produced by this image editing model are entirely determined by user input and actions. The development team and this open-source project are not responsible for any outcomes or consequences arising from its use.

LICENSE

Step1X-Edit is licensed under the Apache License 2.0. You can find the license files in the respective github and HuggingFace repositories.

Name		Name	Last commit message	Last commit date
Latest commit History 79 Commits
GEdit-Bench		GEdit-Bench
assets		assets
examples		examples
library		library
modules		modules
scripts		scripts
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
finetuning.py		finetuning.py
gradio_app.py		gradio_app.py
inference.py		inference.py
requirements.txt		requirements.txt
sampling.py		sampling.py

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Repository files navigation

🔥🔥🔥 News!!

📖 Introduction

⚡️ Quick Start

Step1X-Edit-v1p2 (v1.2)

Step1X-Edit-v1p2-preview (v1.2-preview)

Step1X-Edit-v1p1 (v1.1)

🌟 Advanced Usage

A1. Requirements

A2. Reduce GPU Memory Usage

A3. Multi-GPU inference

A4. Finetuning

Lora training script

Inference with Lora

📊 Benchmark

🧩 Community Contributions

📚 Citation

Acknowledgement

Disclaimer

LICENSE

About

Uh oh!

Releases

Packages

Uh oh!

Contributors 8

Uh oh!

Languages

License

stepfun-ai/Step1X-Edit

Folders and files

Latest commit

History

Repository files navigation

🔥🔥🔥 News!!

📖 Introduction

⚡️ Quick Start

Step1X-Edit-v1p2 (v1.2)

Step1X-Edit-v1p2-preview (v1.2-preview)

Step1X-Edit-v1p1 (v1.1)

🌟 Advanced Usage

A1. Requirements

A2. Reduce GPU Memory Usage

A3. Multi-GPU inference

A4. Finetuning

Lora training script

Inference with Lora

📊 Benchmark

🧩 Community Contributions

📚 Citation

Acknowledgement

Disclaimer

LICENSE

About

Topics

Resources

License

Uh oh!

Stars

Watchers

Forks

Releases

Packages 0

Uh oh!

Contributors 8

Uh oh!

Languages

Packages