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

remove debug print

Browse Source
This commit is contained in:
Kohya S
2023-07-05 21:44:46 +09:00
parent ce46aa0c3b
commit 3060eb5baf
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/train_util.py
View File

@@ -3720,7 +3720,6 @@ def get_noise_noisy_latents_and_timesteps(args, noise_scheduler, latents):
b_size = latents.shape[0]
min_timestep = 0 if args.min_timestep is None else args.min_timestep
max_timestep = noise_scheduler.config.num_train_timesteps if args.max_timestep is None else args.max_timestep
print(b_size, min_timestep, max_timestep)
timesteps = torch.randint(min_timestep, max_timestep, (b_size,), device=latents.device)
timesteps = timesteps.long()