Fixed issue with prompt token replace adding more than one replacement

2026-04-29 02:31:17 +00:00 · 2023-08-26 18:52:23 -06:00
parent fd338e67bb
commit 6bd3851058
2 changed files with 3 additions and 1 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -312,6 +312,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
                if self.trigger_word is not None and not is_reg:
                    prompt = self.sd.inject_trigger_into_prompt(
                        prompt,
+                        trigger=self.trigger_word,
                        add_if_not_present=True,
                    )
                conditioned_prompts.append(prompt)