github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix random_noise_multiplier (#738)

Browse Source 
Apply random_noise_multiplier to noise.
This commit is contained in:
Remix
2026-03-23 21:22:16 +00:00
committed by GitHub
parent 99a4a5887b
commit 7a0bbca5b1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/BaseSDTrainProcess.py
View File

@@ -1358,7 +1358,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
if self.train_config.random_noise_multiplier > 0.0: if self.train_config.random_noise_multiplier > 0.0:
sigma = self.train_config.random_noise_multiplier sigma = self.train_config.random_noise_multiplier
noise_multiplier = torch.exp(torch.randn(s, device=noise.device, dtype=noise.dtype) * sigma) noise_multiplier = torch.exp(torch.randn(s, device=noise.device, dtype=noise.dtype) * sigma)
noise = noise * noise_multiplier
with self.timer('make_noisy_latents'): with self.timer('make_noisy_latents'):
latent_multiplier = self.train_config.latent_multiplier latent_multiplier = self.train_config.latent_multiplier