From 6b7fb60a220fc16f3e7261bd117bfa2b0c87cc49 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 13 Aug 2026 09:52:02 -0600 Subject: [PATCH] Make contrastive guidance loss do constant instead of sigma loss schedule by default --- toolkit/config_modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py index a3a5113..8512eef 100644 --- a/toolkit/config_modules.py +++ b/toolkit/config_modules.py @@ -594,7 +594,7 @@ class TrainConfig: # toward 1.0 as sigma falls (effective = 1 + (target - 1) * sigma) so the # extrapolation never amplifies the unpredictable fresh-noise term at low # sigma. Needed for guidance-distilled models with no guidance embedding. - self.guidance_loss_schedule: str = kwargs.get('guidance_loss_schedule', 'sigma') + self.guidance_loss_schedule: str = kwargs.get('guidance_loss_schedule', 'constant') self.unconditional_prompt: str = kwargs.get('unconditional_prompt', '') if isinstance(self.guidance_loss_target, tuple): self.guidance_loss_target = list(self.guidance_loss_target)