Add replacing triggers on prompts when caching text encoder

2026-07-19 09:03:05 -06:00 · 2026-07-19 09:03:05 -06:00 · 479c72ada2
parent 7ba7e35e19
commit 479c72ada2
1 changed files with 5 additions and 0 deletions
--- 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(