python3 infer_flux_kontext.py \
    --config configs/kontext.nj.yaml \
    --opts \
        flux_path /path/to/FLUX.1-Kontext-dev \
        test.ckpt_path /path/to/checkpoint \
        test.save_path outputs \
        test.json_file data/test_set.json