From 817f3dcbcb357a2f79d38c542ccdc179c365ed8b Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Fri, 7 Aug 2026 09:49:12 -0600 Subject: [PATCH] Fix finite check on inverted masked prior --- extensions_built_in/sd_trainer/SDTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 54c317a..a5197a4 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -939,7 +939,7 @@ class SDTrainer(BaseSDTrainProcess): prior_loss = torch.nn.functional.mse_loss(pred.float(), prior_pred.float(), reduction="none") prior_loss = prior_loss * prior_mask_multiplier * self.train_config.inverted_mask_prior_multiplier - if torch.isnan(prior_loss).any() or not torch.isfinite(prior_loss): + if not torch.isfinite(prior_loss).all(): print_acc("Prior loss is nan") prior_loss = None else: