Fix random_noise_multiplier (#738)

Apply random_noise_multiplier to noise.
2026-03-23 21:22:16 +00:00 · 2026-03-23 21:22:16 +00:00 · 7a0bbca5b1
parent 99a4a5887b
commit 7a0bbca5b1
1 changed files with 1 additions and 1 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@ -1358,7 +1358,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
                if self.train_config.random_noise_multiplier > 0.0:
                    sigma = self.train_config.random_noise_multiplier
                    noise_multiplier = torch.exp(torch.randn(s, device=noise.device, dtype=noise.dtype) * sigma)
-                
+                    noise = noise * noise_multiplier
            with self.timer('make_noisy_latents'):

                latent_multiplier = self.train_config.latent_multiplier