github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Remove reg as prior pred

Browse Source
This commit is contained in:
Jaret Burkett
2024-07-21 02:34:12 -06:00
parent 6e92922c14
commit 8d799031cf
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -1371,9 +1371,9 @@ class SDTrainer(BaseSDTrainProcess):
prior_pred = None prior_pred = None
do_reg_prior = False do_reg_prior = False
if is_reg and (self.network is not None or self.adapter is not None): # 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 # # we are doing a reg image and we have a network or adapter
do_reg_prior = True # do_reg_prior = True
do_inverted_masked_prior = False do_inverted_masked_prior = False
if self.train_config.inverted_mask_prior and batch.mask_tensor is not None: if self.train_config.inverted_mask_prior and batch.mask_tensor is not None: