fix

kohya-ss · Oct 5, 2024 · 77d4e1f · 77d4e1f
1 parent 061ff89
commit 77d4e1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -5239,8 +5239,8 @@ def get_timesteps_and_huber_c(args, min_timestep, max_timestep, noise_scheduler,
             timesteps = time_shift(mu, 1.0, timesteps)
         else:
             timesteps = (timesteps * shift) / (1 + (shift - 1) * timesteps)
-        t = timesteps.view(-1, 1, 1, 1)
         timesteps = min_timestep + (timesteps * (max_timestep - min_timestep))
+        timesteps = timesteps.view(-1, 1, 1, 1)
     else:
         timesteps = torch.randint(min_timestep, max_timestep, (b_size,), device="cpu")