From 4883a1bc4c6faddb41ad0ee91119358f060ae889 Mon Sep 17 00:00:00 2001
From: duongve13112002 <71595470+duongve13112002@users.noreply.github.com>
Date: Wed, 1 Oct 2025 19:40:42 +0700
Subject: [PATCH] Update lumina_train_util.py

---
 library/lumina_train_util.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/library/lumina_train_util.py b/library/lumina_train_util.py
index 32ac8b03..8c621d45 100644
--- a/library/lumina_train_util.py
+++ b/library/lumina_train_util.py
@@ -479,7 +479,6 @@ def time_shift(mu: float, sigma: float, t: torch.Tensor):
     # Since we adopt the reverse, the 1-t operations are needed
     t = 1 - t
     t = math.exp(mu) / (math.exp(mu) + (1 / t - 1) ** sigma)
-    t = 1 - t
     return t
 
 
@@ -854,7 +853,7 @@ def get_noisy_model_input_and_timesteps(
         mu = get_lin_function(y1=0.5, y2=1.15)((h // 2) * (w // 2))
         t = time_shift(mu, 1.0, t)
 
-        timesteps = 1 - t * 1000.0
+        timesteps = t * 1000.0
         t = t.view(-1, 1, 1, 1)
         noisy_model_input = (1 - t) * noise + t * latents
     else: