Step1X-Edit: a unified image editing model performs impressively on various genuine user instructions.
## 🧩 Model Usages
Install the `diffusers` package from the following command:
```bash
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 `Step1XEditPipelineV1P2` class to edit images with thinking and reflection:
```python
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])
```
The results will look like:
