From 8d799031cf7ab8b1524f8671a6b8a5ad6c489ef7 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 21 Jul 2024 02:34:12 -0600 Subject: [PATCH] Remove reg as prior pred --- extensions_built_in/sd_trainer/SDTrainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 48bfcb22..de8aefbc 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -1371,9 +1371,9 @@ class SDTrainer(BaseSDTrainProcess): prior_pred = None do_reg_prior = False - if is_reg and (self.network is not None or self.adapter is not None): - # we are doing a reg image and we have a network or adapter - do_reg_prior = True + # if is_reg and (self.network is not None or self.adapter is not None): + # # we are doing a reg image and we have a network or adapter + # do_reg_prior = True do_inverted_masked_prior = False if self.train_config.inverted_mask_prior and batch.mask_tensor is not None: