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

Fix alphas cumprod after add_noise for DDIMScheduler

Browse Source
This commit is contained in:
rockerBOO
2025-07-15 22:36:46 -04:00
parent 3e6935a07e
commit a7b33f3204
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -6008,6 +6008,8 @@ def get_noise_noisy_latents_and_timesteps(
else:
noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps)
noise_scheduler.alphas_cumprod = noise_scheduler.alphas_cumprod.cpu()
return noise, noisy_latents, timesteps