Flux模型训练脚本