From a5220e9ff303080c8890ca818383abdaa769555e Mon Sep 17 00:00:00 2001 From: Darren Laurie <6DammK9@gmail.com> Date: Sun, 22 Mar 2026 23:02:54 +0800 Subject: [PATCH] fix null pointer --- library/train_util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 3aaba757..5b01ba6c 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -4569,9 +4569,9 @@ def verify_fm_training_args(args: argparse.Namespace): if args.v_pred_like_loss: logger.warning("`--v_pred_like_loss` is ignored when Flow Matching is enabled.") args.v_pred_like_loss = None - if args.flow_use_ot: - logger.info("Using cosine optimal transport pairing for Flow Matching batches.") - raise NotImplementedError("`--flow_use_ot` is not available in this fork.") + # if args.flow_use_ot: + # logger.info("Using cosine optimal transport pairing for Flow Matching batches.") + # raise NotImplementedError("`--flow_use_ot` is not available in this fork.") shift_enabled = args.flow_uniform_shift or args.flow_uniform_static_ratio is not None distribution = args.flow_timestep_distribution