From 3060eb5bafbb4c9c1092fb55d0b7c044b46c045b Mon Sep 17 00:00:00 2001 From: Kohya S Date: Wed, 5 Jul 2023 21:44:46 +0900 Subject: [PATCH] remove debug print --- library/train_util.py | 1 - 1 file changed, 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index 534771ad..9ab1f538 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -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()