Step1X-Edit

🔥🔥🔥 News!!

• Apr 29, 2026: 🎉 Step Image Edit 2 is now live — a lightweight model designed for ultra-fast response and high-quality output, delivering a real-time interactive creation experience. It can complete image generation and editing tasks within 2 seconds. Feel free to try it out and share your feedback ✨✨✨

  Try it here (StepFun Open Platform): https://platform.stepfun.com/docs/zh/guides/models/step-image-edit-2

  API documentation: https://platform.stepfun.com/docs/zh/step-plan/integrations/image-api

• Dec 29, 2025: 🎉 RegionE delivers a 2.5× speedup for Step1X-Edit inference with no accuracy degradation, achieved with just five lines of code.

• 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
  	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.

⚡️ Quick Start

1. Make sure your transformers==4.55.0 (we tested on this version)
2. 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 .

pip install RegionE # optional, for faster inference

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
from RegionE import RegionEHelper

pipe = Step1XEditPipelineV1P2.from_pretrained("stepfun-ai/Step1X-Edit-v1p2", torch_dtype=torch.bfloat16)
pipe.to("cuda")

# Import the RegionEHelper
regionehelper = RegionEHelper(pipe)
regionehelper.set_params()   # default hyperparameter
regionehelper.enable()

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)

regionehelper.disable()

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.