ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

More fixes for noise schedules and fixed targeted guidance inverted masked prior

Browse Source
This commit is contained in:
Jaret Burkett
2023-11-29 10:13:31 -07:00
parent be815f9c47
commit 7624241032
4 changed files with 119 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
jobs/process/BaseSDTrainProcess.py
View File

@@ -775,7 +775,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
if batch.unconditional_latents is not None:
batch.unconditional_latents = batch.unconditional_latents * self.train_config.latent_multiplier
noisy_latents = self.sd.noise_scheduler.add_noise(latents, noise, timesteps)
noisy_latents = self.sd.add_noise(latents, noise, timesteps)
# determine scaled noise
# todo do we need to scale this or does it always predict full intensity