Adjust signal amplification target. Allow signal amplification strength in config.

extensions_built_in/sd_trainer/SDTrainer.py

@@ -578,6 +578,8 @@ class SDTrainer(BaseSDTrainProcess):
                 raise ValueError("Signal amplification is only supported for flow matching models")
             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