From 1cf1f2b94bfa082d2801d6ede5004c45bfd95fff Mon Sep 17 00:00:00 2001 From: araleza <70412719+araleza@users.noreply.github.com> Date: Sun, 24 Aug 2025 16:49:34 +0100 Subject: [PATCH] The optimizer type check in a warning message was incorrect --- flux_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flux_train.py b/flux_train.py index 9bb58c0f..5228ec13 100644 --- a/flux_train.py +++ b/flux_train.py @@ -384,7 +384,7 @@ def train(args): optimizer_train_fn = lambda: None # dummy function optimizer_eval_fn = lambda: None # dummy function - if (args.optimizer_type not in fused_optimizers_supported) and args.full_bf16: + if (args.optimizer_type in fused_optimizers_supported) and args.full_bf16: logger.warning("Use of --blockwise_fused_optimizers is preventing stochastic/Kahan weight updates.") else: _, _, optimizer = train_util.get_optimizer(args, trainable_params=params_to_optimize)