From 479c72ada29219a22c7aa70444e080833a335db1 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 19 Jul 2026 09:03:05 -0600 Subject: [PATCH] Add replacing triggers on prompts when caching text encoder --- extensions_built_in/sd_trainer/SDTrainer.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 6d19c649..f3d07cd1 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -124,6 +124,11 @@ class SDTrainer(BaseSDTrainProcess): for i in range(len(self.sample_config.prompts)): sample_item = self.sample_config.samples[i] prompt = self.sample_config.prompts[i] + + if self.trigger_word is not None: + prompt = self.sd.inject_trigger_into_prompt( + prompt, self.trigger_word, add_if_not_present=False + ) # needed so we can autoparse the prompt to handle flags gen_img_config = GenerateImageConfig(