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

Remove the 0.1 floor for amplification

Browse Source
This commit is contained in:
Jaret Burkett
2026-03-22 09:01:58 -06:00
parent 0f075fc45e
commit 7e1fdc3844
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -579,7 +579,6 @@ class SDTrainer(BaseSDTrainProcess):
with torch.no_grad(): with torch.no_grad():
nas = 1.0 - (timesteps / 1000).to(noise.device, dtype=noise.dtype) nas = 1.0 - (timesteps / 1000).to(noise.device, dtype=noise.dtype)
nas = nas * self.train_config.signal_amplification_strength nas = nas * self.train_config.signal_amplification_strength
nas = nas.clamp(min=0.1)
while len(nas.shape) < len(noise.shape): while len(nas.shape) < len(noise.shape):
nas = nas.unsqueeze(-1) nas = nas.unsqueeze(-1)
aug = batch.latents * nas aug = batch.latents * nas