MrFlow Example Scripts

This directory provides parameterized MrFlow examples for different model families and operating points. The scripts expose prompts, checkpoint paths, random seeds, output directories, and refinement settings through command-line arguments.

Main Settings

Setting	Low-resolution steps	Refinement steps	Refinement sigma	Use case
`12plus1`	12	1	`0.12`	Aggressive acceleration.
`20plus1`	20	1	`0.12`	Higher-quality operating point.

The direct-sigma schedule explicitly specifies the starting noise level of the high-resolution refinement stage.

For flux2_mrflow.py, the available presets are:

Setting	Variant	Low-resolution steps	Refinement steps	Refinement sigma	Guidance scale
`base4b_12plus1`	FLUX.2 Klein Base 4B	12	1	`0.10`	`4.0`
`base9b_12plus1`	FLUX.2 Klein Base 9B	12	1	`0.10`	`4.0`
`4b_4plus1`	FLUX.2 Klein 4B	4	1	`0.25`	`1.0`
`9b_4plus1`	FLUX.2 Klein 9B	4	1	`0.25`	`1.0`

For zimage_turbo_mrflow.py, the default operating point uses --stage1-steps 9, --refine-steps 9, and --strength 0.11. These values are exposed as command-line arguments because Z-Image-Turbo uses its own reduced-step schedule.

Pi-Flow examples require a separate local checkout of LakonLab; it is not vendored in this repository. Clone https://github.com/Lakonik/LakonLab and set LAKONLAB_ROOT to that checkout before running qwen_image_piflow_mrflow.py or flux1_piflow_mrflow.py.

Script Index

Script	Backbone	Notes
`qwen_image_mrflow.py`	Qwen-Image	Training-free MrFlow.
`flux1_mrflow.py`	FLUX.1-dev	Training-free MrFlow.
`qwen_image_piflow_mrflow.py`	Qwen-Image + Pi-Flow	Uses distilled adapter/checkpoint inputs.
`flux1_piflow_mrflow.py`	FLUX.1-dev + Pi-Flow	Uses distilled adapter/checkpoint inputs.
`flux2_mrflow.py`	FLUX.2 Klein	Supports base and non-base settings.
`zimage_turbo_mrflow.py`	Z-Image-Turbo	Adds MrFlow refinement to a reduced-step model.
`direct_sigma_refine.py`	Shared helper	Builds explicit direct-sigma refinement schedules.
`piflow_local.py`	Pi-Flow helper	Local LakonLab import and scheduler shims used by Pi-Flow demos.
`zimage_utils.py`	Z-Image helper	Small wrapper utilities used by the Z-Image-Turbo demo.

Usage

Qwen-Image:

python examples/qwen_image_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${QWEN_IMAGE}" \
  --realesrgan-x2 "${REALESRGAN_X2}" \
  --setting 12plus1

FLUX.1-dev:

python examples/flux1_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${FLUX1_DEV}" \
  --realesrgan-x2 "${REALESRGAN_X2}" \
  --setting 20plus1

Qwen-Image + Pi-Flow:

export LAKONLAB_ROOT="/path/to/LakonLab"
python examples/qwen_image_piflow_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${QWEN_IMAGE}" \
  --adapter-root "${PI_QWEN_ADAPTER_ROOT}" \
  --realesrgan-x2 "${REALESRGAN_X2}"

FLUX.1-dev + Pi-Flow:

export LAKONLAB_ROOT="/path/to/LakonLab"
python examples/flux1_piflow_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${FLUX1_DEV}" \
  --adapter-root "${PI_FLUX_ADAPTER_ROOT}" \
  --realesrgan-x2 "${REALESRGAN_X2}"

FLUX.2 Klein Base:

python examples/flux2_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${FLUX2_KLEIN_BASE_9B}" \
  --realesrgan-x2 "${REALESRGAN_X2}" \
  --setting base9b_12plus1

FLUX.2 Klein non-base:

python examples/flux2_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${FLUX2_KLEIN_9B}" \
  --realesrgan-x2 "${REALESRGAN_X2}" \
  --setting 9b_4plus1

Z-Image-Turbo:

python examples/zimage_turbo_mrflow.py \
  --prompt "${PROMPT}" \
  --model "${Z_IMAGE_TURBO}" \
  --realesrgan-x2 "${REALESRGAN_X2}" \
  --stage1-steps 9 \
  --refine-steps 9 \
  --strength 0.11

You can also edit all placeholder paths in run_examples.sh and run:

bash examples/run_examples.sh

run_examples.sh skips Pi-Flow examples by default because LakonLab and Pi-Flow adapter checkpoints are external. To include them, set:

export LAKONLAB_ROOT="/path/to/LakonLab"
RUN_PIFLOW=1 bash examples/run_examples.sh

Outputs

Root quick-start demos write fixed filenames:

stage1_low.png
stage2_upscaled.png
stage3_refined.png

Parameterized scripts in this directory add a descriptive prefix containing the model family, setting, seed, and resolution, for example:

qwen_image_mrflow_12plus1_seed2026_1024x1024_stage1_low.png
qwen_image_mrflow_12plus1_seed2026_1024x1024_stage2_upscaled.png
qwen_image_mrflow_12plus1_seed2026_1024x1024_stage3_refined.png

The final image is always the stage3_refined file.