Add a config flag to trigger fast image size db builder. Add config flag to set unconditional prompt for guidance loss

2026-04-29 10:41:28 +00:00 · 2025-06-24 08:51:29 -06:00
parent ba1274d99e
commit f3eb1dff42
3 changed files with 15 additions and 10 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -147,7 +147,7 @@ class SDTrainer(BaseSDTrainProcess):
        # cache unconditional embeds (blank prompt)
        with torch.no_grad():
            self.unconditional_embeds = self.sd.encode_prompt(
-                [''],
+                [self.train_config.unconditional_prompt],
                long_prompts=self.do_long_prompts
            ).to(
                self.device_torch,