VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply IP gamma to noise fix

Browse Source
This commit is contained in:
rockerBOO
2025-03-18 15:42:18 -04:00
parent 0b25a05e3c
commit c8be141ae0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/flux_train_utils.py
View File

@@ -471,7 +471,7 @@ def get_noisy_model_input_and_timesteps(
# Add noise according to flow matching.
sigmas = get_sigmas(noise_scheduler, timesteps, device, n_dim=latents.ndim, dtype=dtype)
noisy_model_input = sigmas * noise + (1.0 - sigmas) * latents + ip_noise_gamma
noisy_model_input = sigmas * noise + ip_noise_gamma + (1.0 - sigmas) * latents
return noisy_model_input.to(dtype), timesteps.to(dtype), sigmas