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

Fix typo

Browse Source
This commit is contained in:
Dango233
2024-11-07 18:21:04 +08:00
committed by GitHub
parent e54462a4a9
commit bafd10d558
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sd3_train_utils.py
View File

@@ -980,7 +980,7 @@ def get_noisy_model_input_and_timesteps(
indices = (u * (t_max-t_min) + t_min).long() indices = (u * (t_max-t_min) + t_min).long()
timesteps = indices.to(device=device, dtype=dtype) timesteps = indices.to(device=device, dtype=dtype)
# sigmas according to dlowmatching # sigmas according to flowmatching
sigmas = timesteps / 1000 sigmas = timesteps / 1000
sigmas = sigmas.view(-1,1,1,1) sigmas = sigmas.view(-1,1,1,1)
noisy_model_input = sigmas * noise + (1.0 - sigmas) * latents noisy_model_input = sigmas * noise + (1.0 - sigmas) * latents