Remove the 0.1 floor for amplification

2026-03-22 09:01:58 -06:00
parent 0f075fc45e
commit 7e1fdc3844
1 changed files with 0 additions and 1 deletions
--- 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