From d360e76661eb1b9c48ae3796134c0c332f482ae6 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 26 Feb 2025 13:35:18 -0700 Subject: [PATCH] fixed issue with dop prompt replacement --- extensions_built_in/sd_trainer/SDTrainer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index e22f225c..32f0c2fb 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -1362,7 +1362,9 @@ class SDTrainer(BaseSDTrainProcess): if self.train_config.diff_output_preservation: dop_prompts = [p.replace(self.trigger_word, self.train_config.diff_output_preservation_class) for p in conditioned_prompts] - dop_prompts_2 = [p.replace(self.trigger_word, self.train_config.diff_output_preservation_class) for p in prompt_2] + dop_prompts_2 = None + if prompt_2 is not None: + dop_prompts_2 = [p.replace(self.trigger_word, self.train_config.diff_output_preservation_class) for p in prompt_2] self.diff_output_preservation_embeds = self.sd.encode_prompt( dop_prompts, dop_prompts_2, dropout_prob=self.train_config.prompt_dropout_prob,