From a055947d56a156411abb4ef0301b35d0fe5f51a5 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 7 Feb 2026 12:04:21 -0700 Subject: [PATCH] Add signal_correction_noise_scale to config to scale the signal correction strength --- jobs/process/BaseSDTrainProcess.py | 2 +- toolkit/config_modules.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index 58a3ae67..925d34da 100644 --- a/jobs/process/BaseSDTrainProcess.py +++ b/jobs/process/BaseSDTrainProcess.py @@ -1312,7 +1312,7 @@ class BaseSDTrainProcess(BaseTrainProcess): batch_noise.shape[0], batch_noise.shape[1], 1, 1, device=batch_noise.device, dtype=batch_noise.dtype - ) + ) * self.train_config.signal_correction_noise_scale batch_noise = batch_noise * scn_scale noise = noise + batch_noise diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py index bbe3887d..a48cbe09 100644 --- a/toolkit/config_modules.py +++ b/toolkit/config_modules.py @@ -398,6 +398,7 @@ class TrainConfig: self.target_noise_multiplier = kwargs.get('target_noise_multiplier', 1.0) self.random_noise_multiplier = kwargs.get('random_noise_multiplier', 0.0) self.do_signal_correction_noise = kwargs.get('do_signal_correction_noise', False) + self.signal_correction_noise_scale = kwargs.get('signal_correction_noise_scale', 1.0) self.random_noise_shift = kwargs.get('random_noise_shift', 0.0) self.img_multiplier = kwargs.get('img_multiplier', 1.0) self.noisy_latent_multiplier = kwargs.get('noisy_latent_multiplier', 1.0)