From 7e1fdc3844c7a7f758717fdb6f6aa3b86bf4341b Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 22 Mar 2026 09:01:58 -0600 Subject: [PATCH] Remove the 0.1 floor for amplification --- extensions_built_in/sd_trainer/SDTrainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index 13442d6d..fa31424d 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -579,7 +579,6 @@ class SDTrainer(BaseSDTrainProcess): with torch.no_grad(): nas = 1.0 - (timesteps / 1000).to(noise.device, dtype=noise.dtype) nas = nas * self.train_config.signal_amplification_strength - nas = nas.clamp(min=0.1) while len(nas.shape) < len(noise.shape): nas = nas.unsqueeze(-1) aug = batch.latents * nas