From df354da23e7f89a07780f740bf175efe0383e577 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 20 Jul 2026 07:56:04 -0600 Subject: [PATCH] Replace trigger word in validation prompts for [trigger] tags --- jobs/process/BaseSDTrainProcess.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index dc813b0..a4c2585 100644 --- a/jobs/process/BaseSDTrainProcess.py +++ b/jobs/process/BaseSDTrainProcess.py @@ -1612,7 +1612,14 @@ class BaseSDTrainProcess(BaseTrainProcess): img = transforms.CenterCrop(resolution)(img) tensor = transforms.ToTensor()(img) * 2.0 - 1.0 image_list.append(tensor) - prompt_list.append(item.prompt) + prompt = item.prompt + if self.trigger_word is not None: + prompt = self.sd.inject_trigger_into_prompt( + prompt, + trigger=self.trigger_word, + add_if_not_present=False, + ) + prompt_list.append(prompt) fork_devices = [device] if device.type == 'cuda' else [] with torch.no_grad(), torch.random.fork_rng(devices=fork_devices):