From a4f3a9fc1a4f4f964a6971bc4b0ae15c94f0d672 Mon Sep 17 00:00:00 2001
From: rockerBOO <rockerboo@gmail.com>
Date: Tue, 18 Mar 2025 18:44:21 -0400
Subject: [PATCH] Use ones_like

---
 library/flux_train_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/flux_train_utils.py b/library/flux_train_utils.py
index f0744747..8cf95858 100644
--- a/library/flux_train_utils.py
+++ b/library/flux_train_utils.py
@@ -423,7 +423,7 @@ def get_noisy_model_input_and_timesteps(
         else:
             ip_noise = args.ip_noise_gamma * torch.randn_like(latents)
     else:
-        ip_noise = torch.zeros_like(latents)
+        ip_noise = torch.ones_like(latents)
 
     if args.timestep_sampling == "uniform" or args.timestep_sampling == "sigmoid":
         # Simple random t-based noise sampling