fixed issue with dop prompt replacement

This commit is contained in:
Jaret Burkett committed 2025-02-26 13:35:18 -07:00
1 parent 6ec23ed226
commit d360e76661
1 file changed
+2
@@ -1362,6 +1362,8 @@ 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 = 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,