VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-14 16:22:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

The optimizer type check in a warning message was incorrect

Browse Source
This commit is contained in:
araleza
2025-08-24 16:49:34 +01:00
parent c7b62f7474
commit 1cf1f2b94b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flux_train.py
View File

@@ -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)