FLUX2-klein模型多图输入的训练任务

[shichennianhua]

目前example/flux2中train.py只支持文生图任务的data格式，用data["image"]、data["prompt"]来解析。请问如果是图片输入+prompt的图片编辑任务，以及多图片输入+prompt的风格迁移任务，需要怎么组织数据？是否有一个train.py的样例可以提供？