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

Use ones_like

Browse Source
This commit is contained in:
rockerBOO
2025-03-18 18:44:21 -04:00
parent b425466e7b
commit a4f3a9fc1a
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

@@ -423,7 +423,7 @@ def get_noisy_model_input_and_timesteps(
else: else:
ip_noise = args.ip_noise_gamma * torch.randn_like(latents) ip_noise = args.ip_noise_gamma * torch.randn_like(latents)
else: else:
ip_noise = torch.zeros_like(latents) ip_noise = torch.ones_like(latents)
if args.timestep_sampling == "uniform" or args.timestep_sampling == "sigmoid": if args.timestep_sampling == "uniform" or args.timestep_sampling == "sigmoid":
# Simple random t-based noise sampling # Simple random t-based noise sampling